diff --git a/tests/test_cpu_gpu_equivalence.py b/tests/test_cpu_gpu_equivalence.py
index 49f32314f245f67e8a6b23c5e641c6a4fb933d1b..ae71ca174aa3e29ca7d9bcab3c7927ffc2707bfe 100644
--- a/tests/test_cpu_gpu_equivalence.py
+++ b/tests/test_cpu_gpu_equivalence.py
@@ -16,6 +16,9 @@ def run_equivalence_test(domain_size, lbm_config, lbm_opt, base_config, time_ste
     cpu_scenario = create_channel(domain_size=domain_size, pressure_difference=0.001,
                                   lbm_config=lbm_config, lbm_optimisation=lbm_opt, config=config)
     config = replace(base_config, target=Target.GPU)
+    if not IS_PYSTENCILS_2:
+        from pystencils.enums import Backend
+        config = replace(config, backend=Backend.CUDA)
     gpu_scenario = create_channel(domain_size=domain_size, pressure_difference=0.001,
                                   lbm_config=lbm_config, lbm_optimisation=lbm_opt, config=config)