diff --git a/analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/components/KtFirReferenceShortener.kt b/analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/components/KtFirReferenceShortener.kt
index de2e98a8874..02f2372c70f 100644
--- a/analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/components/KtFirReferenceShortener.kt
+++ b/analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/components/KtFirReferenceShortener.kt
@@ -382,6 +382,12 @@ private class ElementsToShortenCollector(
         }
     }
 
+    override fun visitScript(script: FirScript) {
+        script.statements.forEach {
+            it.accept(this)
+        }
+    }
+
     override fun visitElement(element: FirElement) {
         element.acceptChildren(this)
     }