diff --git a/plugins/hls-class-plugin/test/Main.hs b/plugins/hls-class-plugin/test/Main.hs
index 86bfc33c7c..f9cd09201c 100644
--- a/plugins/hls-class-plugin/test/Main.hs
+++ b/plugins/hls-class-plugin/test/Main.hs
@@ -98,9 +98,8 @@ codeActionTests = testGroup
     action <- head . concatMap (^.. _CACodeAction) <$> getAllCodeActions doc
     executeCodeAction action
     _ <- waitForDiagnostics
-    -- TODO: uncomment this after lsp-test fixed
-    -- ver3 <- (^.J.version) <$> getVersionedDoc doc
-    -- liftIO $ ver3 @?= Just 3
+    ver3 <- (^. L.version) <$> getVersionedDoc doc
+    liftIO $ ver3 @?= 2
     pure mempty
   ]