diff --git a/plugins/lombok/src/test/java/de/plushnikov/intellij/plugin/AbstractLombokParsingTestCase.java b/plugins/lombok/src/test/java/de/plushnikov/intellij/plugin/AbstractLombokParsingTestCase.java
index 90dfd8920102..8020e2637911 100644
--- a/plugins/lombok/src/test/java/de/plushnikov/intellij/plugin/AbstractLombokParsingTestCase.java
+++ b/plugins/lombok/src/test/java/de/plushnikov/intellij/plugin/AbstractLombokParsingTestCase.java
@@ -141,7 +141,7 @@ public abstract class AbstractLombokParsingTestCase extends AbstractLombokLightC
     }
     else if (myRunnerType == ModeRunnerType.INCOMPLETE) {
       IncompleteDependenciesService service = getProject().getService(IncompleteDependenciesService.class);
-      try (var ignored = asAutoCloseable(WriteAction.compute(() -> service.enterIncompleteState()))) {
+      try (var ignored = asAutoCloseable(WriteAction.compute(() -> service.enterIncompleteState(this)))) {
         compareFiles(lowercaseFirstLetter);
       }
     }
diff --git a/plugins/lombok/src/test/java/de/plushnikov/intellij/plugin/highlights/IncompleteModeHighlightTest.java b/plugins/lombok/src/test/java/de/plushnikov/intellij/plugin/highlights/IncompleteModeHighlightTest.java
index d0f5b47260f2..86768949fa2a 100644
--- a/plugins/lombok/src/test/java/de/plushnikov/intellij/plugin/highlights/IncompleteModeHighlightTest.java
+++ b/plugins/lombok/src/test/java/de/plushnikov/intellij/plugin/highlights/IncompleteModeHighlightTest.java
@@ -20,7 +20,7 @@ public class IncompleteModeHighlightTest extends AbstractLombokHighlightsTest {
 
   private void doIncompleteTest() {
     IncompleteDependenciesService service = getProject().getService(IncompleteDependenciesService.class);
-    try (var ignored = asAutoCloseable(WriteAction.compute(() -> service.enterIncompleteState()))) {
+    try (var ignored = asAutoCloseable(WriteAction.compute(() -> service.enterIncompleteState(this)))) {
       String name = getTestName(false);
       myFixture.configureByFile(name + ".java");
       myFixture.testHighlighting(true, true, true);