diff --git a/hls-module-name-plugin.cabal b/hls-module-name-plugin.cabal
--- a/hls-module-name-plugin.cabal
+++ b/hls-module-name-plugin.cabal
@@ -1,6 +1,6 @@
 cabal-version:      2.4
 name:               hls-module-name-plugin
-version:            2.1.0.0
+version:            2.2.0.0
 synopsis:           Module name plugin for Haskell Language Server
 description:
   Please see the README on GitHub at <https://github.com/haskell/haskell-language-server#readme>
@@ -33,8 +33,8 @@
     , containers
     , directory
     , filepath
-    , ghcide                == 2.1.0.0
-    , hls-plugin-api        == 2.1.0.0
+    , ghcide                == 2.2.0.0
+    , hls-plugin-api        == 2.2.0.0
     , lsp
     , text
     , transformers
@@ -53,4 +53,4 @@
     , base
     , filepath
     , hls-module-name-plugin
-    , hls-test-utils          == 2.1.0.0
+    , hls-test-utils          == 2.2.0.0
diff --git a/test/Main.hs b/test/Main.hs
--- a/test/Main.hs
+++ b/test/Main.hs
@@ -38,7 +38,7 @@
       executeCommand c
       void $ skipManyTill anyMessage (message SMethod_WorkspaceApplyEdit)
   , testCase "Should not show code lens if the module name is correct" $
-      runSessionWithServer moduleNamePlugin testDataDir $ do
+      runSessionWithServer def moduleNamePlugin testDataDir $ do
         doc <- openDoc "CorrectName.hs" "haskell"
         lenses <- getCodeLenses doc
         liftIO $ lenses @?= []
@@ -49,7 +49,7 @@
       executeCommand c
       void $ skipManyTill anyMessage (message SMethod_WorkspaceApplyEdit)
   , testCase "Keep stale lens even if parse failed" $ do
-      runSessionWithServer moduleNamePlugin testDataDir $ do
+      runSessionWithServer def moduleNamePlugin testDataDir $ do
         doc <- openDoc "Stale.hs" "haskell"
         oldLens <- getCodeLenses doc
         let edit = TextEdit (mkRange 1 0 1 0) "f ="
@@ -61,7 +61,7 @@
   ]
 
 goldenWithModuleName :: TestName -> FilePath -> (TextDocumentIdentifier -> Session ()) -> TestTree
-goldenWithModuleName title path = goldenWithHaskellDoc moduleNamePlugin title testDataDir path "expected" "hs"
+goldenWithModuleName title path = goldenWithHaskellDoc def moduleNamePlugin title testDataDir path "expected" "hs"
 
 testDataDir :: FilePath
 testDataDir = "test" </> "testdata"