1 files changed, 3 insertions, 3 deletions

diff --git a/src/tpc.y b/src/tpc.y
index 16dc5ad..1e6b6e2 100644
--- a/src/tpc.y
+++ b/src/tpc.y
@@ -135,9 +135,9 @@ Instr:
 | IF '(' Exp IfHandling')' Instr    { gen_if_label($<num>4); }
 | IF '(' Exp IfHandling')' Instr
   ELSE IfEndHandling Instr IfElseEndHandling
-| WHILE                             { fprintf(output,".upwhile%d:\n", num_while); }
-  '(' Exp                           { fprintf(output,"pop rax\ncmp rax,0\njz .downwhile%d\n", num_while); }
-  ')' Instr                         { fprintf(output,"jmp .upwhile%d\n.downwhile%d:\n", num_while, num_while); num_while++; }
+| WHILE                             { gen_while_start(num_while); }
+  '(' Exp                           { gen_while_cond(num_while); }
+  ')' Instr                         { gen_while_end(num_while++); }
 | '{' SuiteInstr '}'
 ;
 IfHandling:                         { gen_if_start($<num>$ = num_if++); };