From e26100ffa35f86633c1b8728f4cf0d547bc486b1 Mon Sep 17 00:00:00 2001
From: IlyaMuravjov <muravjovilya@gmail.com>
Date: Fri, 28 Jul 2023 14:51:30 +0300
Subject: [PATCH] Handle `modelTagName` for `UtCustomModel`

---
 .../domain/models/builders/SpringTestClassModelBuilder.kt     | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/builders/SpringTestClassModelBuilder.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/builders/SpringTestClassModelBuilder.kt
index 49dc213fcf..8294a09e94 100644
--- a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/builders/SpringTestClassModelBuilder.kt
+++ b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/builders/SpringTestClassModelBuilder.kt
@@ -113,7 +113,9 @@ class SpringTestClassModelBuilder(val context: CgContext): TestClassModelBuilder
                 is UtClassRefModel,
                 is UtVoidModel,
                 is UtEnumConstantModel -> {}
-                is UtCustomModel -> currentModel.origin?.let { collectRecursively(it.wrap(), allModels) }
+                is UtCustomModel -> currentModel.origin?.let {
+                    collectRecursively(it.wrap(currentModelWrapper.modelTagName), allModels)
+                }
                 is UtLambdaModel -> {
                     currentModel.capturedValues.forEach { collectRecursively(it.wrap(), allModels) }
                 }