From 0f6e2a3b65bc33af9cadd7ded7cbeed098e46960 Mon Sep 17 00:00:00 2001
From: Kirill Shishin <shishin_ka@mail.ru>
Date: Thu, 3 Aug 2023 16:25:05 +0300
Subject: [PATCH] Add instanceModel to dependentModel

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

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 4bb35e2ad3..edf3ea6593 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
@@ -91,6 +91,8 @@ class SpringTestClassModelBuilder(val context: CgContext) :
     private fun collectDependentModels(model: UtModel): Set<UtModelWrapper> {
         val dependentModels = mutableSetOf<UtModelWrapper>()
 
+        dependentModels.add(model.wrap())
+
         when (model) {
             is UtNullModel,
             is UtPrimitiveModel,