diff --git a/idea/testData/inspections/unusedImport/suppression.kt b/idea/testData/inspections/unusedImport/suppression.kt
index cd6684b02b0..4ccf8064b31 100644
--- a/idea/testData/inspections/unusedImport/suppression.kt
+++ b/idea/testData/inspections/unusedImport/suppression.kt
@@ -1,3 +1,4 @@
 @file:Suppress("UnusedImport")
+package suppression
 
-import java.io.* // unused
\ No newline at end of file
+import java.io.* // unused
diff --git a/idea/tests/org/jetbrains/kotlin/idea/codeInsight/AbstractInspectionTest.kt b/idea/tests/org/jetbrains/kotlin/idea/codeInsight/AbstractInspectionTest.kt
index 03e034eb022..ea61c50879a 100644
--- a/idea/tests/org/jetbrains/kotlin/idea/codeInsight/AbstractInspectionTest.kt
+++ b/idea/tests/org/jetbrains/kotlin/idea/codeInsight/AbstractInspectionTest.kt
@@ -94,7 +94,7 @@ abstract class AbstractInspectionTest : KotlinLightCodeInsightFixtureTestCase()
                     file.extension == "kt" -> {
                         val text = FileUtil.loadFile(file, true)
                         val fileText =
-                            if (text.startsWith("package"))
+                            if (text.lines().any { it.startsWith("package") })
                                 text
                             else
                                 "package ${file.nameWithoutExtension};$text"