diff --git a/src/pairs/code_gen/cgen.py b/src/pairs/code_gen/cgen.py
index 5d3c9dbc3d662f92f816c6aeebdd083e206b74fe..516479b7ef75a1d9aa014ac13f2b68f88108001f 100644
--- a/src/pairs/code_gen/cgen.py
+++ b/src/pairs/code_gen/cgen.py
@@ -404,7 +404,7 @@ class CGen:
                 module_params += decl if len(module_params) <= 0 else f", {decl}"
 
             for var in module.write_variables():
-                decl = f"&{var.name()}"
+                decl = f"rv_{var.name()}.getDevicePointer()" if device_cond and var.device_flag else f"&{var.name()}"
                 module_params += decl if len(module_params) <= 0 else f", {decl}"
 
             for array in module.arrays():