Fix Travis build

2 files changed, 3 insertions, 2 deletions

diff --git a/tests/Main.hs b/tests/Main.hs
index 6577e6f4..75321ec7 100644
--- a/tests/Main.hs
+++ b/tests/Main.hs
@@ -10,6 +10,7 @@ import System.Environment
 import System.Process
 import System.Directory
 import Text.Printf
+import Test.Hspec
 
 import Config
 import Spec
@@ -64,7 +65,7 @@ terminateEnv = do
 runTestSuite :: [String] -> IO ExitCode
 runTestSuite args = do
   printf "running hspec test suite with args: %s\n" (show args)
-  catch (withArgs args hspecMain >> return ExitSuccess) return
+  catch (withArgs args (hspec spec) >> return ExitSuccess) return
 
 main :: IO ()
 main = do
diff --git a/tests/Spec.hs b/tests/Spec.hs
index 8d2b5139..b4e92e75 100644
--- a/tests/Spec.hs
+++ b/tests/Spec.hs
@@ -1 +1 @@
-{-# OPTIONS_GHC -F -pgmF hspec-discover -optF --hook-main #-}
+{-# OPTIONS_GHC -F -pgmF hspec-discover -optF --no-main #-}