From 65b85388b9c55ea3c6ddf8af650abec8c66c5ec9 Mon Sep 17 00:00:00 2001
From: Konstantin Chukharev <lipen00@gmail.com>
Date: Mon, 7 Jul 2025 19:35:37 +0300
Subject: [PATCH] Change 'arg' in AwaitExpr to be Local

---
 jacodb-ets/src/main/kotlin/org/jacodb/ets/dto/Convert.kt | 2 +-
 jacodb-ets/src/main/kotlin/org/jacodb/ets/model/Expr.kt  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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 {