move return into the right place

diff --git a/src/comp.c b/src/comp.c
index 3f0db646660260e01f1bef08ee3b933ee1b73933..4b01a057054a29d63bee6beec90e7332c8eaadf0 100644 (file)
--- a/src/comp.c
+++ b/src/comp.c
@@ -763,6 +763,10 @@ compile_f (const char *f_name, ptrdiff_t bytestr_length,
          break;
 
        case Breturn:
+         POP1;
+         gcc_jit_block_end_with_return(comp.block,
+                                       NULL,
+                                       args[0]);
          break;
 
        case Bdiscard:
@@ -930,10 +934,6 @@ compile_f (const char *f_name, ptrdiff_t bytestr_length,
        }
     }
 
-  stack--;
-  gcc_jit_block_end_with_return(comp.block,
-                               NULL,
-                               *stack);
   comp_res.gcc_res = gcc_jit_context_compile(comp.ctxt);
 
   goto exit;