diff --git a/libs/jit/source/arch/aarch64/compiler_v2.cpp b/libs/jit/source/arch/aarch64/compiler_v2.cpp
index 69ceefb..a52e1ad 100644
--- a/libs/jit/source/arch/aarch64/compiler_v2.cpp
+++ b/libs/jit/source/arch/aarch64/compiler_v2.cpp
@@ -258,6 +258,9 @@ namespace pslang::jit::aarch64
 				else
 					throw std::runtime_error(std::format("Unsupported load size: {}", size));
 
+				if (types::is_bool_type(*type) || types::is_integer_type(*type))
+					extend(0, type);
+
 				store(it, 0);
 			}