diff --git a/twasm/asm/main.asm b/twasm/asm/main.asm
index b958731..239e0d5 100644
--- a/twasm/asm/main.asm
+++ b/twasm/asm/main.asm
@@ -90,7 +90,7 @@ start:
 ; ------------------------------------------------------------------------------
 
 assemble:
-  ; TODO deal with src=imm and src=imm8
+  mov rbp, rsp
   xor eax, eax
   mov [.tokens_processed], eax ; eax = number of tokens processed
   mov [.tokens_total], edi     ; edi = total number of tokens in table
@@ -940,7 +940,8 @@ assemble:
     mov rsi, .msg_break
     call print.debug
     pop rsi
-    jmp halt
+    mov rsp, rbp
+    ret
 
   .unexpected_token:
     call .flush_write_buffer