diff --git a/tests/unit/src/main/scala/tests/BaseWorksheetLspSuite.scala b/tests/unit/src/main/scala/tests/BaseWorksheetLspSuite.scala
index 89adb6812af..e8446facf2e 100644
--- a/tests/unit/src/main/scala/tests/BaseWorksheetLspSuite.scala
+++ b/tests/unit/src/main/scala/tests/BaseWorksheetLspSuite.scala
@@ -420,7 +420,8 @@ abstract class BaseWorksheetLspSuite(
       _ <- server.didChange("a/src/main/scala/a/Util.scala")(
         _.replace("n + 1", "n + 2")
       )
-      _ <- server.didSave("a/src/main/scala/a/Main.worksheet.sc")
+      _ <- server.didSave("a/src/main/scala/a/Util.scala")
+      _ <- server.didFocus("a/src/main/scala/a/Main.worksheet.sc")
       _ <- server.assertInlayHints(
         "a/src/main/scala/a/Main.worksheet.sc",
         """
diff --git a/tests/unit/src/test/scala/tests/TreeViewLspSuite.scala b/tests/unit/src/test/scala/tests/TreeViewLspSuite.scala
index 54413b6eceb..4615d7c2ead 100644
--- a/tests/unit/src/test/scala/tests/TreeViewLspSuite.scala
+++ b/tests/unit/src/test/scala/tests/TreeViewLspSuite.scala
@@ -176,7 +176,6 @@ class TreeViewLspSuite extends BaseLspSuite("tree-view") {
       _ <- server.didChange("a/src/main/scala/a/Zero.scala") { text =>
         text.replace("val a = 1", "val a = 1\nval b = 1.0")
       }
-      _ <- server.didSave("a/src/main/scala/a/Zero.scala")
       _ = assertEquals(
         server.client.workspaceTreeViewChanges,
         s"metalsPackages projects-$folder:${server.buildTarget("a")}!/_root_/",