diff --git a/libs/jit/source/arch/aarch64/compiler.cpp b/libs/jit/source/arch/aarch64/compiler.cpp
index 16d8802..b0e7c08 100644
--- a/libs/jit/source/arch/aarch64/compiler.cpp
+++ b/libs/jit/source/arch/aarch64/compiler.cpp
@@ -198,7 +198,8 @@ namespace pslang::jit::aarch64
 
 			void apply(ast::return_statement const & node)
 			{
-				apply(*node.value);
+				if (node.value)
+					apply(*node.value);
 			}
 
 			void apply(ast::field_definition const &)
@@ -817,7 +818,8 @@ namespace pslang::jit::aarch64
 
 			void apply(ast::return_statement const & node)
 			{
-				apply(*node.value);
+				if (node.value)
+					apply(*node.value);
 				if (stack_offset > 0)
 					builder.add_imm(31, 31, stack_offset);
 				builder.ret();