diff --git a/HW4Part1.hs b/HW4Part1.hs
index 3c09f4c..7bf8a32 100644
--- a/HW4Part1.hs
+++ b/HW4Part1.hs
@@ -32,7 +32,7 @@ program =
     ]
 
 while :: Expr -> Prog -> Stmt
-while cond body = Loop $ If cond [] [ Break ] : body
+while cond body = Loop [ If cond body [ Break ] ]
 
 sumFromTo :: Int -> Int -> Prog
 sumFromTo f t =