diff --git a/heat/dxarray/tests/test_dxarray.py b/heat/dxarray/tests/test_dxarray.py
index 22517ac506..4c93b3abf0 100644
--- a/heat/dxarray/tests/test_dxarray.py
+++ b/heat/dxarray/tests/test_dxarray.py
@@ -18,7 +18,7 @@ def test_dim_name_idx_conversion(self):
             idxs = ht.dxarray.dim_name_to_idx(dims, names)
             # check for correct types (str, tuple or list)
             self.assertTrue(
-                type(idxs) == type(names) or (isinstance(names, str) and isinstance(idxs, int))
+                type(idxs) is type(names) or (isinstance(names, str) and isinstance(idxs, int))
             )
             # check if dim_name_to_idx and dim_idx_to_name are inverse to each other
             names_back = ht.dxarray.dim_idx_to_name(dims, idxs)