diff --git a/src/main/wacc/backend/asmGenerator.scala b/src/main/wacc/backend/asmGenerator.scala
index 16b9a41..4c1a924 100644
--- a/src/main/wacc/backend/asmGenerator.scala
+++ b/src/main/wacc/backend/asmGenerator.scala
@@ -25,6 +25,7 @@ object asmGenerator {
         funcPrologue() ++
         alignStack() ++
         main.flatMap(generateStmt) ++
+        List(Move(Register(RegSize.R64, RegName.AX), ImmediateVal(0))) ++
         funcEpilogue() ++
         List(assemblyIR.Return()) ++
         generateFuncs()
@@ -204,7 +205,6 @@ object asmGenerator {
 
   def funcEpilogue(): List[AsmLine] = {
     List(
-      Move(Register(RegSize.R64, RegName.AX), ImmediateVal(0)),
       Move(Register(RegSize.R64, RegName.SP), Register(RegSize.R64, RegName.BP)),
       Pop(Register(RegSize.R64, RegName.BP))
     )