diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/GenerateTestsDialogWindow.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/GenerateTestsDialogWindow.kt
index d831f47cdb..9e621d1fad 100644
--- a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/GenerateTestsDialogWindow.kt
+++ b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/GenerateTestsDialogWindow.kt
@@ -1107,7 +1107,7 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
 
         mockStrategies.addActionListener { _ ->
             updateControlsEnabledStatus()
-            if (staticsMocking.isSelected && mockStrategies.item == MockStrategyApi.NO_MOCKS) {
+            if (mockStrategies.item == MockStrategyApi.NO_MOCKS) {
                 staticsMocking.isSelected = false
             }
         }
@@ -1156,7 +1156,7 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
                 profileNames.text = ""
             }
 
-            if (!staticsMocking.isSelected && isSpringConfigSelected()) {
+            if (isSpringConfigSelected() && springTestType.item == UNIT_TEST) {
                 staticsMocking.isSelected = true
             }