diff --git a/analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/utils/shortenReferencesUtils.kt b/analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/utils/shortenReferencesUtils.kt
index 1c0174deeee..28b0a9262da 100644
--- a/analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/utils/shortenReferencesUtils.kt
+++ b/analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/utils/shortenReferencesUtils.kt
@@ -59,7 +59,10 @@ fun addImportToFile(
     val newDirective = psiFactory.createImportDirective(importPath)
     val imports = importList.imports
     if (imports.isEmpty()) {
-        importList.add(psiFactory.createNewLine(2))
+        file.packageDirective?.takeIf { it.packageKeyword != null }?.let {
+            file.addAfter(psiFactory.createNewLine(2), it)
+        }
+
         importList.add(newDirective)
     } else {
         val insertAfter = imports.lastOrNull {