1 files changed, 4 insertions, 1 deletions

diff --git a/test/TestSundqvist.hs b/test/TestSundqvist.hs
index 25d3f8f..d02da10 100644
--- a/test/TestSundqvist.hs
+++ b/test/TestSundqvist.hs
@@ -1,9 +1,11 @@
 module TestSundqvist where
 
 
+import Data.Maybe
 import Graphics.WaveFront
 import Graphics.WaveFront.Load as Load
 import System.Directory
+import System.Environment
 import System.IO
 
 getDataDir :: IO FilePath
@@ -15,5 +17,6 @@ getDataDir = do
 
 main = do
     ddir <- getDataDir
-    r1 <- Load.obj (ddir ++ "/01.obj")
+    fn <- fromMaybe (ddir ++ "/01.obj") . listToMaybe <$>  getArgs
+    r1 <- Load.obj fn
     either print (mapM_ print) r1