diff --git a/2020/day01/day01.hs b/2020/day01/day01.hs
index c98a12c..33123c4 100644
--- a/2020/day01/day01.hs
+++ b/2020/day01/day01.hs
@@ -4,10 +4,10 @@ import Control.Monad
 main :: IO ()
 main = do
   input <- map read . lines <$> readFile "input"
-  print $ day1 input 2
-  print $ day1 input 3
+  print $ day1 2 input
+  print $ day1 3 input
 
-day1 :: [Int] -> Int -> Int
-day1 input n =
-  snd . head . filter ((== 2020) . fst) $
-  map (sum &&& product) $ replicateM n input
+day1 :: Int -> [Int] -> Int
+day1 =
+  fmap (snd . head . filter ((== 2020) . fst) . map (sum &&& product)) .
+  replicateM