diff --git a/HW4.idr b/HW4.idr
index 33f3e26..c7f91bd 100644
--- a/HW4.idr
+++ b/HW4.idr
@@ -38,7 +38,9 @@ initialState = (IntTy, IntTy, IntTy)
 testProg : Prog Main.initialState Main.initialState
 testProg =
   [ Store A (Lit 1 `Leq` Lit 2)
-  , Store A (Lit 1)
+  , If (Load A)
+    [ Store A (Lit 1) ]
+    [ Store A (Lit 2) ]
   , Store B (Lit 2)
   , Store R (Add (Load A) (Load B))
   ]