diff --git a/python/python-venv/tests/com/intellij/python/junit5Tests/env/venv/showCase/PyEnvWithVenvShowCaseTest.kt b/python/python-venv/tests/com/intellij/python/junit5Tests/env/venv/showCase/PyEnvWithVenvShowCaseTest.kt
index 857939781435..3737024304ef 100644
--- a/python/python-venv/tests/com/intellij/python/junit5Tests/env/venv/showCase/PyEnvWithVenvShowCaseTest.kt
+++ b/python/python-venv/tests/com/intellij/python/junit5Tests/env/venv/showCase/PyEnvWithVenvShowCaseTest.kt
@@ -14,7 +14,7 @@ import org.junit.jupiter.api.Test
 @PyEnvTestCase
 class PyEnvWithVenvShowCaseTest {
   private val tempPathFixture = tempPathFixture()
-  private val moduleFixture = projectFixture().moduleFixture()
+  private val moduleFixture = projectFixture().moduleFixture(tempPathFixture, addPathToSourceRoot = true)
   private val venvFixture = pySdkFixture().pyVenvFixture( // <-- venv fixture
     where = tempPathFixture,
     addToSdkTable = true,
diff --git a/python/src/com/jetbrains/python/projectCreation/venvWithSdkCreator.kt b/python/src/com/jetbrains/python/projectCreation/venvWithSdkCreator.kt
index 2a75fb16f92d..4a15d828c1a5 100644
--- a/python/src/com/jetbrains/python/projectCreation/venvWithSdkCreator.kt
+++ b/python/src/com/jetbrains/python/projectCreation/venvWithSdkCreator.kt
@@ -21,8 +21,8 @@ import com.jetbrains.python.errorProcessing.failure
 import com.jetbrains.python.sdk.configurePythonSdk
 import com.jetbrains.python.sdk.createSdk
 import com.jetbrains.python.sdk.flavors.PythonSdkFlavor
-import com.jetbrains.python.sdk.getOrCreateAdditionalData
 import com.jetbrains.python.sdk.pythonSdk
+import com.jetbrains.python.sdk.setAssociationToModuleAsync
 import com.jetbrains.python.venvReader.VirtualEnvReader
 import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.withContext
@@ -86,7 +86,7 @@ suspend fun createVenvAndSdk(
     VfsUtil.markDirtyAndRefresh(false, true, true, vfsProjectPath)
   }
   configurePythonSdk(project, module, sdk)
-  sdk.getOrCreateAdditionalData().associateWithModule(module)
+  sdk.setAssociationToModuleAsync(module)
   module.pythonSdk
   return Result.success(sdk)
 }