diff --git a/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/dsl/ForLoopBody.kt b/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/dsl/ForLoopBody.kt
index 264cb62f815b..08b62aa61bd0 100644
--- a/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/dsl/ForLoopBody.kt
+++ b/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/dsl/ForLoopBody.kt
@@ -20,5 +20,5 @@ package com.intellij.debugger.streams.trace.dsl
  */
 interface ForLoopBody : CodeBlock {
   val loopVariable: Variable
-  fun breakIteration()
+  fun breakIteration(): Expression
 }
\ No newline at end of file
diff --git a/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/dsl/impl/java/JavaForLoopBody.kt b/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/dsl/impl/java/JavaForLoopBody.kt
index 8f01c0a739a0..3cdb2a90bcba 100644
--- a/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/dsl/impl/java/JavaForLoopBody.kt
+++ b/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/dsl/impl/java/JavaForLoopBody.kt
@@ -15,6 +15,7 @@
  */
 package com.intellij.debugger.streams.trace.dsl.impl.java
 
+import com.intellij.debugger.streams.trace.dsl.Expression
 import com.intellij.debugger.streams.trace.dsl.ForLoopBody
 import com.intellij.debugger.streams.trace.dsl.StatementFactory
 import com.intellij.debugger.streams.trace.dsl.Variable
@@ -29,5 +30,5 @@ class JavaForLoopBody(statementFactory: StatementFactory, override val loopVaria
     val BREAK = TextExpression("break")
   }
 
-  override fun breakIteration() = addStatement(BREAK)
+  override fun breakIteration(): Expression = BREAK
 }
\ No newline at end of file
diff --git a/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/dsl/impl/kotlin/KotlinForLoopBody.kt b/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/dsl/impl/kotlin/KotlinForLoopBody.kt
index 8212b3cf03bd..2b3f2449e3f0 100644
--- a/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/dsl/impl/kotlin/KotlinForLoopBody.kt
+++ b/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/dsl/impl/kotlin/KotlinForLoopBody.kt
@@ -15,10 +15,7 @@
  */
 package com.intellij.debugger.streams.trace.dsl.impl.kotlin
 
-import com.intellij.debugger.streams.trace.dsl.Dsl
-import com.intellij.debugger.streams.trace.dsl.ForLoopBody
-import com.intellij.debugger.streams.trace.dsl.StatementFactory
-import com.intellij.debugger.streams.trace.dsl.Variable
+import com.intellij.debugger.streams.trace.dsl.*
 import com.intellij.debugger.streams.trace.dsl.impl.TextExpression
 
 /**
@@ -30,7 +27,5 @@ class KotlinForLoopBody(override val loopVariable: Variable,
     val BREAK = TextExpression("break")
   }
 
-  override fun breakIteration() {
-    addStatement(BREAK)
-  }
+  override fun breakIteration(): Expression = BREAK
 }
\ No newline at end of file