diff --git a/test/contracts/list_comp.aes b/test/contracts/list_comp.aes
index 266f2e8..db98bb5 100644
--- a/test/contracts/list_comp.aes
+++ b/test/contracts/list_comp.aes
@@ -10,11 +10,11 @@ contract ListComp =
     entrypoint l2_true() = [5,6,6,7,7,8]
 
     entrypoint l3() = [x ++ y | x <- [[":)"] | x <- [1,2]]
-                              , y <- [[":("]]]  
+                              , y <- [[":("]]]
     entrypoint l3_true() = [[":)", ":("], [":)", ":("]]
 
     entrypoint l4() = [(a, b, c) | let is_pit(a, b, c) = a*a + b*b == c*c
-                                 , let base = [1,2,3,4,5,6,7,8,9,10]
+                                 , let base = [1..10]
                                  , a <- base
                                  , b <- base, if (b >= a)
                                  , c <- base, if (c >= b)