diff --git a/pystencils/rng.py b/pystencils/rng.py
index 863867bfed56a8dd4bc3192148fe510c32160109..7c4f894f9871e350fd9a5f531708d123dcb7be2b 100644
--- a/pystencils/rng.py
+++ b/pystencils/rng.py
@@ -5,7 +5,6 @@ import sympy as sp
 from pystencils.data_types import TypedSymbol, cast_func
 from pystencils.astnodes import LoopOverCoordinate
 from pystencils.backends.cbackend import CustomCodeNode
-from pystencils.enums import Backend
 from pystencils.sympyextensions import fast_subs
 
 
diff --git a/pystencils_tests/test_datahandling.py b/pystencils_tests/test_datahandling.py
index f160f5f59a1556e3ebb6efb1844f7b23c2632c56..be695d078384e678d93f2116b5932379035e878a 100644
--- a/pystencils_tests/test_datahandling.py
+++ b/pystencils_tests/test_datahandling.py
@@ -357,12 +357,10 @@ def test_load_data():
     assert np.all(dh.cpu_arrays['dst2']) == 0
 
 
-@pytest.mark.parametrize('target', (Target.GPU))
-def test_array_handler(target):
+def test_array_handler():
     size = (2, 2)
-    if target == Target.GPU:
-        pytest.importorskip('pycuda')
-        array_handler = PyCudaArrayHandler()
+    pytest.importorskip('pycuda')
+    array_handler = PyCudaArrayHandler()
 
     zero_array = array_handler.zeros(size)
     cpu_array = np.empty(size)