diff --git a/app/Main.hs b/app/Main.hs
index 739e0c5..8dd1ea1 100644
--- a/app/Main.hs
+++ b/app/Main.hs
@@ -22,10 +22,11 @@ main :: IO ()
 main =
     defaultMain
         . localOption (Always :: UseColor)
-        $ testGroup
-            "tests"
-            [ puzzleTest puzzle1
-            , puzzleTest puzzle2
+        . testGroup "tests"
+        $ map
+            puzzleTest
+            [ puzzle1
+            , puzzle2
             ]
 
 puzzleTest :: Puzzle -> TestTree