diff --git a/haskell-tools-demo.cabal b/haskell-tools-demo.cabal
--- a/haskell-tools-demo.cabal
+++ b/haskell-tools-demo.cabal
@@ -1,5 +1,5 @@
 name:                haskell-tools-demo
-version:             1.0.0.1
+version:             1.0.0.2
 synopsis:            A web-based demo for Haskell-tools Refactor.
 description:         Allows websocket clients to connect and performs refactorings on demand. The clients maintain a continous connection with the server, sending changes in the source files. When a refactor request is received, it performs the changes and sends the modified source files to the client.
 homepage:            https://github.com/haskell-tools/haskell-tools
@@ -54,8 +54,8 @@
   main-is:             Main.hs
   build-depends:       base                      >= 4.10 && < 4.11
                      , HUnit                     >= 1.5 && < 1.7
-                     , tasty                     >= 0.11 && < 0.12
-                     , tasty-hunit               >= 0.9 && < 0.10
+                     , tasty                     >= 0.11 && < 0.13
+                     , tasty-hunit               >= 0.9 && < 0.11
                      , directory                 >= 1.2 && < 1.4
                      , filepath                  >= 1.4 && < 2.0
                      , bytestring                >= 0.10 && < 0.11
diff --git a/src/Language/Haskell/Tools/Demo.hs b/src/Language/Haskell/Tools/Demo.hs
--- a/src/Language/Haskell/Tools/Demo.hs
+++ b/src/Language/Haskell/Tools/Demo.hs
@@ -55,7 +55,7 @@
 type ClientId = Int
 
 data RefactorSessionState
-  = RefactorSessionState { _refSessMods :: Map.Map (String, String, FilePath) (UnnamedModule IdDom)
+  = RefactorSessionState { _refSessMods :: Map.Map (String, String, FilePath) UnnamedModule
                          , _actualMod :: Maybe (String, String, FilePath)
                          , _isDisconnecting :: Bool
                          }