diff --git a/tests/nerf/test_sampler.py b/tests/nerf/test_sampler.py
index 8e884a4be..167bda16b 100644
--- a/tests/nerf/test_sampler.py
+++ b/tests/nerf/test_sampler.py
@@ -21,7 +21,7 @@ def test_uniform_sampler():
 
     ray_samples = sampler(ray_bundle)
 
-    assert ray_samples.bins.shape[-1] == num_samples + 1
+    assert ray_samples.positions.shape[-2] == num_samples
 
     # TODO Tancik: Add more precise tests