diff --git a/CHANGELOG.md b/CHANGELOG.md
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Change Log
 
+## 0.1.1
+
+* Small bug fix for "style" test when using the cabal distribution
+
 ## 0.1 (initial release)
 
 * Initial release with all Template components implemented along with a huge
diff --git a/README.md b/README.md
--- a/README.md
+++ b/README.md
@@ -1,6 +1,6 @@
 # Stratosphere: AWS CloudFormation in Haskell
 
-[![Build Status](https://travis-ci.org/frontrowed/stratosphere.svg?branch=master)](https://travis-ci.org/frontrowed/stratosphere)
+[![Circle CI](https://circleci.com/gh/frontrowed/stratosphere.svg?style=svg)](https://circleci.com/gh/frontrowed/stratosphere)
 
 AWS CloudFormation is a system that provisions and updates Amazon Web Services
 (AWS) resources based on declarative templates. Common criticisms of
diff --git a/stratosphere.cabal b/stratosphere.cabal
--- a/stratosphere.cabal
+++ b/stratosphere.cabal
@@ -1,9 +1,9 @@
--- This file has been generated from package.yaml by hpack version 0.12.0.
+-- This file has been generated from package.yaml by hpack version 0.14.0.
 --
 -- see: https://github.com/sol/hpack
 
 name:           stratosphere
-version:        0.1.0
+version:        0.1.1
 synopsis:       EDSL for AWS CloudFormation
 description:    EDSL for AWS CloudFormation
 category:       AWS, Cloud
@@ -165,6 +165,7 @@
     , text >= 1.1
     , unordered-containers >= 0.2
     , base
+    , directory
     , hlint
   other-modules:
       Main
diff --git a/tests/HLint.hs b/tests/HLint.hs
--- a/tests/HLint.hs
+++ b/tests/HLint.hs
@@ -1,16 +1,21 @@
 module Main (main) where
 
 import Language.Haskell.HLint (hlint)
+import System.Directory (doesDirectoryExist)
 import System.Exit (exitFailure, exitSuccess)
 
 arguments :: [String]
 arguments =
     [ "library"
     , "tests"
-    , "gen"
     ]
 
 main :: IO ()
 main = do
-    hints <- hlint arguments
+    -- Need to check for existence of gen directory, because it is not present
+    -- in the cabal distribution.
+    genExists <- doesDirectoryExist "gen"
+    let args = if genExists then "gen" : arguments else arguments
+
+    hints <- hlint args
     if null hints then exitSuccess else exitFailure