diff --git a/utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js/JsDialogProcessor.kt b/utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js/JsDialogProcessor.kt
index c770587677..3a1f9bbfd1 100644
--- a/utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js/JsDialogProcessor.kt
+++ b/utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js/JsDialogProcessor.kt
@@ -186,13 +186,16 @@ object JsDialogProcessor {
                 val generatedCode = testGenerator.run()
                 invokeLater {
                     runWriteAction {
-                        val testPsiFile = testDir.findFile(testFileName) ?: PsiFileFactory.getInstance(project)
-                            .createFileFromText(testFileName, JsLanguageAssistant.jsLanguage, generatedCode)
+                        val testPsiFile = testDir.findFile(testFileName) ?: run {
+                            val temp = PsiFileFactory.getInstance(project)
+                                .createFileFromText(testFileName, JsLanguageAssistant.jsLanguage, generatedCode)
+                            testDir.add(temp)
+                            testDir.findFile(testFileName)!!
+                        }
                         val testFileEditor = CodeInsightUtil.positionCursor(project, testPsiFile, testPsiFile)
                         unblockDocument(project, testFileEditor.document)
                         testFileEditor.document.setText(generatedCode)
                         unblockDocument(project, testFileEditor.document)
-                        testDir.findFile(testFileName) ?: testDir.add(testPsiFile)
                     }
                 }
             }