diff --git a/utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonDialogProcessor.kt b/utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonDialogProcessor.kt
index 08311194ed..cd3f348184 100644
--- a/utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonDialogProcessor.kt
+++ b/utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonDialogProcessor.kt
@@ -25,7 +25,6 @@ import kotlinx.coroutines.runBlocking
 import org.jetbrains.kotlin.idea.util.application.runWriteAction
 import org.jetbrains.kotlin.idea.util.module
 import org.jetbrains.kotlin.idea.util.projectStructure.sdk
-import org.jetbrains.kotlin.konan.file.File
 import org.utbot.common.PathUtil.toPath
 import org.utbot.common.appendHtmlLine
 import org.utbot.framework.UtSettings
@@ -335,7 +334,8 @@ fun getDirectoriesForSysPath(
     file.fromImports.forEach { importTarget ->
         importTarget.resolveImportSourceCandidates().forEach {
             val directory = it.parent
-            if (directory is PsiDirectory ) {
+            val isRelativeImport = importTarget.relativeLevel > 0  // If we have `from . import a` we don't need to add syspath
+            if (directory is PsiDirectory && !isRelativeImport) {
                 // If we have `from a.b.c import d` we need to add syspath to module `a` only
                 val additionalLevel = importTarget.importSourceQName?.componentCount?.dec() ?: 0
                 directory.topParent(additionalLevel)?.let { dir ->
@@ -347,7 +347,9 @@ fun getDirectoriesForSysPath(
 
     // Select modules only from this project but not from installation directory
     importedPaths.forEach {
-        if (it.isProjectSubmodule(ancestor) && !it.path.split(File.separator).contains("site-packages")) {
+        val path = it.toNioPath()
+        val hasSitePackages = (0 until(path.nameCount)).any { i -> path.subpath(i, i+1).toString() == "site-packages"}
+        if (it.isProjectSubmodule(ancestor) && !hasSitePackages) {
             sources.add(it)
         }
     }