diff --git a/libs/jit/source/arch/aarch64/compiler.cpp b/libs/jit/source/arch/aarch64/compiler.cpp
index e3d6206..313c23e 100644
--- a/libs/jit/source/arch/aarch64/compiler.cpp
+++ b/libs/jit/source/arch/aarch64/compiler.cpp
@@ -935,7 +935,10 @@ namespace pslang::jit::aarch64
 			void scope_cleanup()
 			{
 				if (scopes.back().stack_offset > 0)
+				{
 					builder.add_imm(31, 31, scopes.back().stack_offset);
+					stack_offset -= scopes.back().stack_offset;
+				}
 			}
 		};