diff --git a/idea/idea-jvm/src/org/jetbrains/kotlin/idea/scratch/output/InlayScratchOutputHandler.kt b/idea/idea-jvm/src/org/jetbrains/kotlin/idea/scratch/output/InlayScratchOutputHandler.kt
index f5518fe7d8a..a99c0a3516c 100644
--- a/idea/idea-jvm/src/org/jetbrains/kotlin/idea/scratch/output/InlayScratchOutputHandler.kt
+++ b/idea/idea-jvm/src/org/jetbrains/kotlin/idea/scratch/output/InlayScratchOutputHandler.kt
@@ -96,10 +96,10 @@ object InlayScratchOutputHandler : ScratchOutputHandler {
 
     private fun maxLineLength(file: ScratchFile): Int {
         val doc = file.editor.editor.document
-        return (0 until doc.lineCount)
+        return file.getExpressions()
+            .flatMap { it.lineStart..it.lineEnd }
             .map { doc.getLineEndOffset(it) - doc.getLineStartOffset(it) }
-            .max()
-                ?: -1
+            .max() ?: 0
     }
 
     private fun clearInlays(editor: TextEditor) {
diff --git a/idea/testData/scratch/unresolved.repl.after b/idea/testData/scratch/unresolved.repl.after
index 5e7af6ce157..ad634ea7927 100644
--- a/idea/testData/scratch/unresolved.repl.after
+++ b/idea/testData/scratch/unresolved.repl.after
@@ -1,4 +1,4 @@
-foo()                                                    // ERROR: error: unresolved reference: foo...
+foo()    // ERROR: error: unresolved reference: foo...
 /** unresolved.kts:1 error: unresolved reference: foo
 
 foo()
diff --git a/idea/testData/scratch/unresolvedMultiline.repl.after b/idea/testData/scratch/unresolvedMultiline.repl.after
index 9e379afdf3e..b6c15da7f85 100644
--- a/idea/testData/scratch/unresolvedMultiline.repl.after
+++ b/idea/testData/scratch/unresolvedMultiline.repl.after
@@ -1,8 +1,8 @@
-foo.forEach {                                               // ERROR: error: unresolved reference: foo...
+foo.forEach {           // ERROR: error: unresolved reference: foo...
     1 + 1
 }
 
-fun goo(a: String) {                                        // ERROR: error: unresolved reference: goo...
+fun goo(a: String) {    // ERROR: error: unresolved reference: goo...
     super.goo(a)
 }
 /** unresolvedMultiline.kts:1 error: unresolved reference: foo