diff --git a/test/test_expression.py b/test/test_expression.py
index 49456ac2f9c3736b43519e49fad6f4f421c6753f..8b74852a680e6ec16304902e02f3bfe6b8ab1fa5 100644
--- a/test/test_expression.py
+++ b/test/test_expression.py
@@ -557,8 +557,9 @@ def test_complex_support(ctx_factory, target):
 
     kwargs = {"in1": in1, "in2": in2}
 
+    knl = lp.set_options(knl, write_code=True)
+
     if target == lp.PyOpenCLTarget:
-        knl = lp.set_options(knl, write_code=True)
         cl_ctx = ctx_factory()
         with cl.CommandQueue(cl_ctx) as queue:
             evt, out = knl(queue, **kwargs)