diff --git a/Codegen.jai b/Codegen.jai
index 8df5e78..8dbe59d 100644
--- a/Codegen.jai
+++ b/Codegen.jai
@@ -627,7 +627,7 @@ emit_declaration :: (state : *Codegen_State, node : *AST_Node) {
 	}
 }
 
-codegen :: (result : *Compiler_Context) {
+codegen :: (result : *Compiler_Context, allocator := temp) {
 	codegen(result, .HLSL);
 }
 
diff --git a/module.jai b/module.jai
index 881df97..bc3aa0f 100644
--- a/module.jai
+++ b/module.jai
@@ -507,10 +507,10 @@ compile_file :: (ctx : *Compiler_Context, path : string, allocator : Allocator =
 
 		ctx.file = make_file(ctx, path);
 
-		lex(ctx);
-		parse(ctx);
-		check(ctx);
-		codegen(ctx);
+		lex(ctx, allocator);
+		parse(ctx, allocator);
+		check(ctx, allocator);
+		codegen(ctx, allocator);
 		generate_output_data(ctx);
 	}
 }