diff --git a/include/aidge/backend/cpu/data/TensorImpl.hpp b/include/aidge/backend/cpu/data/TensorImpl.hpp
index 984e5c945f427e0fea0ea1dc240cedb62fda07d3..02f1ee5f87b7ba81b371898497df2f2eef4f6d4a 100644
--- a/include/aidge/backend/cpu/data/TensorImpl.hpp
+++ b/include/aidge/backend/cpu/data/TensorImpl.hpp
@@ -21,7 +21,7 @@ class TensorImpl_cpu : public TensorImpl {
    public:
     static constexpr const char *Backend = "cpu";
 
-    TensorImpl_cpu(int device, NbElts_t length) : TensorImpl(Backend, device, length) {}
+    TensorImpl_cpu(DeviceIdx_t device, NbElts_t length) : TensorImpl(Backend, device, length) {}
 
     bool operator==(const TensorImpl &otherImpl) const override final {
         const auto& typedOtherImpl = reinterpret_cast<const TensorImpl_cpu<T> &>(otherImpl);
@@ -35,7 +35,7 @@ class TensorImpl_cpu : public TensorImpl {
         return i == mNbElts;
     }
 
-    static std::shared_ptr<TensorImpl_cpu> create(int device, NbElts_t length) {
+    static std::shared_ptr<TensorImpl_cpu> create(DeviceIdx_t device, NbElts_t length) {
         return std::make_shared<TensorImpl_cpu<T>>(device, length);
     }