diff --git a/test/test_response.py b/test/test_response.py
index bb5104aa17fa02c5334fb5ee6016faaba9937e50..07a97b544cdfb5097fb1dd13faf4527e2b907752 100644
--- a/test/test_response.py
+++ b/test/test_response.py
@@ -84,7 +84,7 @@ def test_jax_response_consistency():
     obs = obs.to_double_precision()
 
     sdom = dom[-1]
-    radio_sky = ift.from_random(dom.domain)
+    radio_sky = ift.from_random(dom)
     radio_sky_arr = radio_sky.val
 
     R_old = rve.InterferometryResponse(
@@ -104,4 +104,4 @@ def test_jax_response_consistency():
     vis_field_old = R_old(radio_sky).val
     vis_field_new = R_new(radio_sky_arr)
 
-    np.allclose(vis_field_old, vis_field_new)
+    np.testing.assert_allclose(vis_field_old, vis_field_new)