diff --git a/python/src/com/jetbrains/python/testing/PyTestsShared.kt b/python/src/com/jetbrains/python/testing/PyTestsShared.kt
index 99248c7ffbce..70625456c9a7 100644
--- a/python/src/com/jetbrains/python/testing/PyTestsShared.kt
+++ b/python/src/com/jetbrains/python/testing/PyTestsShared.kt
@@ -625,6 +625,7 @@ object PyTestsConfigurationProducer : com.jetbrains.python.testing.AbstractPytho
     }
 
     val location = context?.location
+    configuration.module = context?.module
     if (location is com.jetbrains.python.testing.PyTargetBasedPsiLocation) {
       location.target.copyTo(configuration.target)
     }
@@ -637,7 +638,6 @@ object PyTestsConfigurationProducer : com.jetbrains.python.testing.AbstractPytho
         configuration.workingDirectory = targetForConfig.second
       }
     }
-    configuration.module = context?.module
     configuration.setGeneratedName()
     return true
   }