1 files changed, 3 insertions, 3 deletions
diff --git a/hypsrc-test/run.hs b/hypsrc-test/run.hs
index 0b97a075..549f33f7 100755
--- a/hypsrc-test/run.hs
+++ b/hypsrc-test/run.hs
@@ -39,9 +39,9 @@ main = do
 
     (args, mods) <- partition ("-" `isPrefixOf`) <$> getArgs
     let args' = filter (\arg -> not $ arg == "--all" || arg == "-a") args
-    mods' <- map (srcDir </>) <$> if "--all" `elem` args || "-a" `elem` args
-        then getAllSrcModules
-        else return mods
+    mods' <- map (srcDir </>) <$> case args of
+        [] -> getAllSrcModules
+        _ -> return $ map (++ ".hs") mods
 
     putHaddockVersion
     putGhcVersion