diff --git a/aidge_backend_cpu/unit_tests/test_tensor.py b/aidge_backend_cpu/unit_tests/test_tensor.py
index 830b5cb56801eff2be62efb40d3b14180ba12d48..37531b43cf7755dfb760e575450b70bfa9a6ff68 100644
--- a/aidge_backend_cpu/unit_tests/test_tensor.py
+++ b/aidge_backend_cpu/unit_tests/test_tensor.py
@@ -59,7 +59,7 @@ class test_tensor(unittest.TestCase):
     def test_get_set(self):
         dims = [2,2,2]
 
-        np_array = np.arange(8).reshape(dims)
+        np_array = np.arange(8).reshape(dims).astype(np.int32)
         # Numpy -> Tensor
         t = aidge_core.Tensor(np_array)
         for i in range(8):