diff --git a/test/test_response.py b/test/test_response.py
index 07a97b544cdfb5097fb1dd13faf4527e2b907752..fbb156f31b065a1dc78ff31fba9365e5d9862480 100644
--- a/test/test_response.py
+++ b/test/test_response.py
@@ -15,8 +15,6 @@
 # Copyright(C) 2022 Max-Planck-Society, Philipp Arras
 # Author: Philipp Arras
 
-from os.path import join
-
 import nifty8 as ift
 import numpy as np
 import pytest
@@ -24,10 +22,6 @@ import pytest
 import resolve as rve
 import resolve.re as jrve
 
-from .common import setup_function, teardown_function
-
-import configparser
-
 pmp = pytest.mark.parametrize
 np.seterr(all="raise")
 
@@ -84,11 +78,14 @@ def test_jax_response_consistency():
     obs = obs.to_double_precision()
 
     sdom = dom[-1]
-    radio_sky = ift.from_random(dom)
+    fdom = rve.IRGSpace([np.mean(OBS[0].freq)])
+    pdom = rve.PolarizationSpace(("I", "Q", "U", "V"))
+    pol_dom = rve.default_sky_domain(sdom=sdom, fdom=fdom, pdom=pdom)
+    radio_sky = ift.from_random(pol_dom)
     radio_sky_arr = radio_sky.val
 
     R_old = rve.InterferometryResponse(
-        obs, dom, do_wgridding=True, epsilon=1e-9, nthreads=1
+        obs, pol_dom, do_wgridding=True, epsilon=1e-9, nthreads=1
     )
     sky_domain_dict = dict(
         npix_x=sdom.shape[0],