tweak codegen

diff --git a/src/prolog/codegen.rs b/src/prolog/codegen.rs
index acd9407eaee5fd6950a89b19d3fc882350f0d66f..abe274e01f3e0e3b946d29cf15a9b2f380fd000e 100644 (file)
--- a/src/prolog/codegen.rs
+++ b/src/prolog/codegen.rs
@@ -256,17 +256,19 @@ impl<'a, TermMarker: Allocator<'a>> CodeGenerator<'a, TermMarker>
                            body: &mut Code)
                            -> usize
     {
-        let perm_vars = vs.vars_above_threshold(0) + deep_cuts as usize;
-
         if num_clauses > 0 {
+            let perm_vars = vs.vars_above_threshold(0) + deep_cuts as usize;
+
             body.push(Line::Control(ControlInstruction::Allocate(perm_vars)));
 
             if deep_cuts {
                 body.push(Line::Cut(CutInstruction::GetLevel));
             }
+
+            return perm_vars;
         }
 
-        perm_vars
+        0
     }
 
     fn compile_neck_cut_or(&mut self,