diff --git a/include/aidge/backend/cuda/data/TensorImpl.hpp b/include/aidge/backend/cuda/data/TensorImpl.hpp
index dfd347dc53ff019048d36b401443335acb2c1f9d..18a6a574f2b5970ab2b828cf69fe66165f52bbbb 100644
--- a/include/aidge/backend/cuda/data/TensorImpl.hpp
+++ b/include/aidge/backend/cuda/data/TensorImpl.hpp
@@ -189,6 +189,7 @@ public:
     }
 
     void *rawPtr(NbElts_t offset = 0) override {
+        cudaSetDevice(mDevice);
         lazyInit();
         return (mData.data() + offset);
     };