diff --git a/README.md b/README.md
--- a/README.md
+++ b/README.md
@@ -3,6 +3,7 @@
 Autoexporter automatically re-exports Haskell modules.
 
 [![Version badge][]][version]
+[![Build badge]][build]
 
 -   [Install](#install)
 -   [Use](#use)
@@ -73,6 +74,8 @@
         this as well.
 
 [Autoexporter]: https://github.com/tfausak/autoexporter
-[Version badge]: https://www.stackage.org/package/autoexporter/badge/nightly?label=version
-[version]: https://www.stackage.org/package/autoexporter
+[Version badge]: https://img.shields.io/hackage/v/autoexporter.svg?label=version
+[version]: https://hackage.haskell.org/package/autoexporter
+[Build badge]: https://travis-ci.org/tfausak/autoexporter.svg?branch=master
+[build]: https://travis-ci.org/tfausak/autoexporter
 [Stack]: http://haskellstack.org
diff --git a/autoexporter.cabal b/autoexporter.cabal
--- a/autoexporter.cabal
+++ b/autoexporter.cabal
@@ -1,5 +1,5 @@
 name: autoexporter
-version: 0.1.0
+version: 0.1.1
 cabal-version: >=1.10
 build-type: Simple
 license: MIT
diff --git a/library/Autoexporter.hs b/library/Autoexporter.hs
--- a/library/Autoexporter.hs
+++ b/library/Autoexporter.hs
@@ -12,7 +12,7 @@
         [name, inputFile, outputFile] -> do
             input <- readFile inputFile
             case input of
-                "{-# OPTIONS_GHC -F -pgmF nafta #-}\n" -> do
+                "{-# OPTIONS_GHC -F -pgmF autoexporter #-}\n" -> do
                     let moduleName = makeModuleName name
 
                     let directory = FilePath.dropExtension inputFile