diff --git a/test/FileMappingSpec.hs b/test/FileMappingSpec.hs
index 91576a3..ccd39be 100644
--- a/test/FileMappingSpec.hs
+++ b/test/FileMappingSpec.hs
@@ -88,7 +88,7 @@ spec = do
             res <- run defaultOptions {fileMappings = fm} $ do
               loadMappedFiles
               checkSyntax ["File.hs"]
-            res `shouldBe` "File.hs:1:1:Warning: Top-level binding with no type signature: main :: forall t. t\n"
+            res `shouldBe` "File.hs:1:1:Warning: Top-level binding with no type signature: main :: IO ()\n"
         it "checks in-memory file if one is specified and outputs original filename" $ do
           withDirectory_ "test/data/file-mapping" $ do
             let fm = [("File.hs", MemoryMapping $ Just "main = putStrLn \"Hello World!\"\n")]
diff --git a/test/data/file-mapping/File_Redir.hs b/test/data/file-mapping/File_Redir.hs
index 45846a9..d5e55cc 100644
--- a/test/data/file-mapping/File_Redir.hs
+++ b/test/data/file-mapping/File_Redir.hs
@@ -1 +1 @@
-main = undefined
+main = putStrLn "Hello World!"