diff --git a/shake-ats.cabal b/shake-ats.cabal
--- a/shake-ats.cabal
+++ b/shake-ats.cabal
@@ -1,5 +1,5 @@
 name:                shake-ats
-version:             0.3.0.5
+version:             0.3.0.6
 synopsis:            Utilities for building ATS projects with shake
 description:         Various helper functions for building [ATS](http://www.ats-lang.org/) with the [shake](http://shakebuild.com/) library
 homepage:            https://github.com/vmchale/shake-ats#readme
@@ -25,7 +25,7 @@
   build-depends:       base >= 4.7 && < 5
                      , language-ats
                      , shake-ext
-                     , hs2ats
+                     , hs2ats >= 0.2.0.0
                      , directory
                      , text
                      , shake
diff --git a/src/Development/Shake/ATS.hs b/src/Development/Shake/ATS.hs
--- a/src/Development/Shake/ATS.hs
+++ b/src/Development/Shake/ATS.hs
@@ -45,8 +45,7 @@
 genATS src target =
     target %> \out -> liftIO $ do
         createDirectoryIfMissing True (takeDirectory out)
-        contents <- readFile src
-        writeFile out (generateATS contents)
+        genATSTypes src target
 
 ghcExport :: String -> Rules ()
 ghcExport m =