diff --git a/source/tests/pt/model/test_polarizability_fitting.py b/source/tests/pt/model/test_polarizability_fitting.py
index a061780f45..84d6bd91ab 100644
--- a/source/tests/pt/model/test_polarizability_fitting.py
+++ b/source/tests/pt/model/test_polarizability_fitting.py
@@ -269,12 +269,7 @@ def test_permu(self):
 
     def test_trans(self):
         atype = self.atype.reshape(1, 5)
-        coord_s = torch.matmul(
-            torch.remainder(
-                torch.matmul(self.coord + self.shift, torch.linalg.inv(self.cell)), 1.0
-            ),
-            self.cell,
-        )
+        coord_s = self.coord + self.shift
         for fit_diag, scale in itertools.product([True, False], [None, self.scale]):
             ft0 = PolarFittingNet(
                 self.nt,