diff --git a/tests/test_points_to_volumes.py b/tests/test_points_to_volumes.py
index 75b0a7c8..c72deeb9 100644
--- a/tests/test_points_to_volumes.py
+++ b/tests/test_points_to_volumes.py
@@ -354,7 +354,7 @@ class TestPointsToVolumes(TestCaseMixin, unittest.TestCase):
                         # this does not produce grads w.r.t. xyz
                         self.assertIsNone(field.grad)
                     else:
-                        self.assertTrue(field.grad.data.isfinite().all())
+                        self.assertTrue(torch.isfinite(field.grad.data).all())
 
     def _check_volume_slice_color_density(
         self, V, split_dim, interp_mode, clr_gt, slice_type, border=3
diff --git a/tests/test_raymarching.py b/tests/test_raymarching.py
index 9e3a64bc..9b77b01b 100644
--- a/tests/test_raymarching.py
+++ b/tests/test_raymarching.py
@@ -163,7 +163,7 @@ class TestRaymarching(TestCaseMixin, unittest.TestCase):
         loss = features_render.mean()
         loss.backward()
         for field in (rays_densities, rays_features):
-            self.assertTrue(field.grad.data.isfinite().all())
+            self.assertTrue(torch.isfinite(field.grad.data).all())
 
     def test_absorption_only(self):
         """
@@ -193,4 +193,4 @@ class TestRaymarching(TestCaseMixin, unittest.TestCase):
         # check differentiability
         loss = opacities_render.mean()
         loss.backward()
-        self.assertTrue(rays_densities.grad.data.isfinite().all())
+        self.assertTrue(torch.isfinite(rays_densities.grad.data).all())