diff --git a/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/impl/handler/unified/PeekTraceHandler.kt b/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/impl/handler/unified/PeekTraceHandler.kt
index 716c23388a4c..2b5d20a97f53 100644
--- a/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/impl/handler/unified/PeekTraceHandler.kt
+++ b/plugins/stream-debugger/src/main/java/com/intellij/debugger/streams/trace/impl/handler/unified/PeekTraceHandler.kt
@@ -45,21 +45,17 @@ class PeekTraceHandler(num: Int, callName: String, private val myTypeBefore: Gen
   }
 
   override fun additionalCallsBefore(): List<IntermediateStreamCall> {
-    val lambda = dsl.code {
-      lambda("x") {
-        +beforeMap.set(currentTime(), +argName)
-      }
-    }
+    val lambda = dsl.lambda("x") {
+      +beforeMap.set(dsl.currentTime(), TextExpression(argName))
+    }.toCode()
 
     return listOf(dsl.createPeekCall(myTypeBefore, lambda))
   }
 
   override fun additionalCallsAfter(): List<IntermediateStreamCall> {
-    val lambda = dsl.code {
-      lambda("x") {
-        +afterMap.set(currentTime(), +argName)
-      }
-    }
+    val lambda = dsl.lambda("x") {
+      +afterMap.set(dsl.currentTime(), TextExpression(argName))
+    }.toCode()
 
     return listOf(dsl.createPeekCall(myTypeBefore, lambda))
   }