diff --git a/platform/lang-impl/src/com/intellij/ide/scratch/RootType.java b/platform/lang-impl/src/com/intellij/ide/scratch/RootType.java
index 12b4d65e26c4..5eb9fbcd990b 100644
--- a/platform/lang-impl/src/com/intellij/ide/scratch/RootType.java
+++ b/platform/lang-impl/src/com/intellij/ide/scratch/RootType.java
@@ -15,9 +15,9 @@
  */
 package com.intellij.ide.scratch;
 
-import com.intellij.ide.util.treeView.AbstractTreeBuilder;
 import com.intellij.lang.Language;
 import com.intellij.lang.LanguageUtil;
+import com.intellij.openapi.Disposable;
 import com.intellij.openapi.extensions.ExtensionPointName;
 import com.intellij.openapi.extensions.Extensions;
 import com.intellij.openapi.fileEditor.FileEditorManager;
@@ -118,7 +118,7 @@ public abstract class RootType {
     return false;
   }
 
-  public void registerTreeUpdater(@NotNull Project project, @NotNull AbstractTreeBuilder builder) {
+  public void registerTreeUpdater(@NotNull Project project, @NotNull Disposable disposable, @NotNull Runnable onUpdate) {
   }
 
 }
diff --git a/platform/lang-impl/src/com/intellij/ide/scratch/ScratchProjectViewPane.java b/platform/lang-impl/src/com/intellij/ide/scratch/ScratchProjectViewPane.java
index 36ce0697ed8b..f44f6a225960 100644
--- a/platform/lang-impl/src/com/intellij/ide/scratch/ScratchProjectViewPane.java
+++ b/platform/lang-impl/src/com/intellij/ide/scratch/ScratchProjectViewPane.java
@@ -106,7 +106,7 @@ public class ScratchProjectViewPane extends ProjectViewPane {
       };
     for (RootType rootId : RootType.getAllRootIds()) {
       if (rootId.isHidden()) continue;
-      rootId.registerTreeUpdater(myProject, builder);
+      rootId.registerTreeUpdater(myProject, builder, builder::queueUpdate);
     }
     return builder;
   }