diff --git a/jacodb-ets/src/main/kotlin/org/jacodb/ets/dto/Convert.kt b/jacodb-ets/src/main/kotlin/org/jacodb/ets/dto/Convert.kt
index ddf297fc7..75ea43bcc 100644
--- a/jacodb-ets/src/main/kotlin/org/jacodb/ets/dto/Convert.kt
+++ b/jacodb-ets/src/main/kotlin/org/jacodb/ets/dto/Convert.kt
@@ -278,7 +278,7 @@ class EtsMethodBuilder(
         )
 
         is AwaitExprDto -> EtsAwaitExpr(
-            arg = arg.toEtsEntity(),
+            arg = ensureLocal(arg.toEtsEntity()),
             type = type.toEtsType(),
         )
 
diff --git a/jacodb-ets/src/main/kotlin/org/jacodb/ets/model/Expr.kt b/jacodb-ets/src/main/kotlin/org/jacodb/ets/model/Expr.kt
index 3fa7f405d..1b7d1bc00 100644
--- a/jacodb-ets/src/main/kotlin/org/jacodb/ets/model/Expr.kt
+++ b/jacodb-ets/src/main/kotlin/org/jacodb/ets/model/Expr.kt
@@ -218,7 +218,7 @@ data class EtsDeleteExpr(
 }
 
 data class EtsAwaitExpr(
-    override val arg: EtsEntity,
+    override val arg: EtsLocal,
     override val type: EtsType,
 ) : EtsUnaryExpr {
     override fun toString(): String {