1 files changed, 4 insertions, 1 deletions

diff --git a/test/TestSabadie.hs b/test/TestSabadie.hs
index d8d1949..76d6aee 100644
--- a/test/TestSabadie.hs
+++ b/test/TestSabadie.hs
@@ -1,6 +1,8 @@
 module TestSabadie where
 
 import Codec.Wavefront
+import Data.Maybe
+import System.Environment
 import System.Directory
 import System.IO
 import Text.Show.Pretty (ppShow)
@@ -14,5 +16,6 @@ getDataDir = do
 
 main = do
     ddir <- getDataDir
-    r1 <- fromFile (ddir ++ "/01.obj")
+    fn <- fromMaybe (ddir ++ "/01.obj") . listToMaybe <$>  getArgs
+    r1 <- fromFile fn
     either print (putStrLn . ppShow) r1