diff --git a/Setup.hs b/Setup.hs
--- a/Setup.hs
+++ b/Setup.hs
@@ -1,2 +1,12 @@
+import Distribution.Simple
 import LiquidHaskell.Cabal
-main = liquidHaskellMain
+
+data Choice = Simple | Post
+
+choice :: Choice
+choice = Post -- Simple
+
+main :: IO ()
+main = case choice of
+         Simple -> liquidHaskellMain
+         Post   -> liquidHaskellMainHooks
diff --git a/liquidhaskell-cabal-demo.cabal b/liquidhaskell-cabal-demo.cabal
--- a/liquidhaskell-cabal-demo.cabal
+++ b/liquidhaskell-cabal-demo.cabal
@@ -1,8 +1,8 @@
 name:                     liquidhaskell-cabal-demo
-version:                  0.1.1.0
+version:                  0.2.0.0
 synopsis:                 Demo of Liquid Haskell integration for Cabal and stack
 description:              Please see the
-                          <https://github.com/spinda/liquidhaskell-cabal-demo/blob/0.1.1.0/README.md README>
+                          <https://github.com/spinda/liquidhaskell-cabal-demo/blob/0.2.0.0/README.md README>
                           on GitHub for more information.
 homepage:                 https://github.com/spinda/liquidhaskell-cabal-demo#readme
 bug-reports:              https://github.com/spinda/liquidhaskell-cabal-demo/issues
@@ -10,15 +10,13 @@
 license-file:             LICENSE
 author:                   Michael Smith
 maintainer:               Michael Smith <michael@spinda.net>
-copyright:                2016 Michael Smith
+copyright:                2016-2019 Michael Smith, 2019 Ranjit Jhala
 category:                 Development
 cabal-version:            >=1.10
 build-type:               Custom
 
--- Unfortunately this won't work until Cabal 1.24 is released.
---
--- custom-setup
---   setup-depends:           base, Cabal, liquidhaskell-cabal
+custom-setup
+  setup-depends:           base, Cabal, liquidhaskell-cabal >= 0.2.0
 
 flag liquidhaskell
   description:             After building, verify with LiquidHaskell
@@ -27,19 +25,17 @@
 library
   exposed-modules:         A, B, C
   build-depends:           base >=4.8 && <5
-                         , liquidhaskell-cabal >= 0.1.1 && < 0.2
+                         , liquidhaskell-cabal >= 0.2
   hs-source-dirs:          src
   default-language:        Haskell2010
   default-extensions:      TupleSections
-  ghc-options:             -rtsopts
-                           -with-rtsopts=-N
-                           -Wall
+  ghc-options:             -Wall
   x-liquidhaskell-options: --diff --no-termination
 
 executable ffi
   main-is:                 FFI.hs
   build-depends:           base >=4.8 && <5
-                         , liquidhaskell-cabal >= 0.1.1 && < 0.2
+                         , liquidhaskell-cabal >= 0.2
   hs-source-dirs:          app
   default-language:        Haskell2010
   default-extensions:      ForeignFunctionInterface
diff --git a/src/A.hs b/src/A.hs
--- a/src/A.hs
+++ b/src/A.hs
@@ -5,5 +5,5 @@
 plus x y = x + y
 
 test :: String -> (String, String)
-test = ("test", )
+test x = ("test", x)
 
diff --git a/src/C.hs b/src/C.hs
--- a/src/C.hs
+++ b/src/C.hs
@@ -6,4 +6,3 @@
 {-@ quux :: x:Int -> y:Int -> z:Int -> {v:Int | v = x + y - z} @-}
 quux :: Int -> Int -> Int -> Int
 quux x y z = x `plus` y `minus` z
-