diff --git a/idea/idea-completion/testData/handlers/multifile/TopLevelFunctionImport-1.kt b/idea/idea-completion/testData/handlers/multifile/TopLevelFunctionImport-1.kt
index df0b5b797d9..d8bb0dfeb84 100644
--- a/idea/idea-completion/testData/handlers/multifile/TopLevelFunctionImport-1.kt
+++ b/idea/idea-completion/testData/handlers/multifile/TopLevelFunctionImport-1.kt
@@ -1,7 +1,6 @@
+// FIR_COMPARISON
 package some
 
-import java.util.SortedSet
-
 fun other() {
     somefu<caret>
 }
diff --git a/idea/idea-completion/testData/handlers/multifile/TopLevelFunctionImport.kt.after b/idea/idea-completion/testData/handlers/multifile/TopLevelFunctionImport.kt.after
index db0b284fc4a..5530f8531b7 100644
--- a/idea/idea-completion/testData/handlers/multifile/TopLevelFunctionImport.kt.after
+++ b/idea/idea-completion/testData/handlers/multifile/TopLevelFunctionImport.kt.after
@@ -1,7 +1,7 @@
+// FIR_COMPARISON
 package some
 
 import jettesting.data.somefun
-import java.util.SortedSet
 
 fun other() {
     somefun(<caret>)