diff --git a/src/pystencils/codegen/driver.py b/src/pystencils/codegen/driver.py
index 74a07b902a6494d0c528602326286a2aaab01cc5..59e3139133ab76097bd1e8febdae2ffb91eea41f 100644
--- a/src/pystencils/codegen/driver.py
+++ b/src/pystencils/codegen/driver.py
@@ -320,8 +320,7 @@ class DefaultKernelCreationDriver:
         append_ast = [PsAssignment(ptr_access, write_back_ptr)]
 
         # modify AST
-        kernel_ast.statements = prepend_ast + kernel_ast.statements
-        kernel_ast.statements += append_ast
+        kernel_ast.statements = prepend_ast + kernel_ast.statements + append_ast
 
     def _transform_for_cpu(self, kernel_ast: PsBlock) -> PsBlock:
         canonicalize = CanonicalizeSymbols(self._ctx, True)