diff --git a/plugins/evaluation-plugin/core/src/com/intellij/cce/test/TestRunner.kt b/plugins/evaluation-plugin/core/src/com/intellij/cce/test/TestRunner.kt
index a4d937bb4c6d..0afacbb5e935 100644
--- a/plugins/evaluation-plugin/core/src/com/intellij/cce/test/TestRunner.kt
+++ b/plugins/evaluation-plugin/core/src/com/intellij/cce/test/TestRunner.kt
@@ -6,7 +6,7 @@ import com.intellij.openapi.project.Project
 
 class TestRunnerParams(val language: Language)
 
-class TestRunRequest(val tests: List<String>, val project: Project)
+class TestRunRequest(val tests: List<String>, val testCommand: String?, val project: Project)
 class TestRunResult(
   val exitCode: Int,
   val passed: List<String>,
diff --git a/plugins/evaluation-plugin/languages/python/src/com/intellij/cce/python/test/PythonTestRunner.kt b/plugins/evaluation-plugin/languages/python/src/com/intellij/cce/python/test/PythonTestRunner.kt
index afe11e6e2928..a3de881f6361 100644
--- a/plugins/evaluation-plugin/languages/python/src/com/intellij/cce/python/test/PythonTestRunner.kt
+++ b/plugins/evaluation-plugin/languages/python/src/com/intellij/cce/python/test/PythonTestRunner.kt
@@ -16,7 +16,7 @@ class PythonTestRunner : TestRunner {
 
   override suspend fun runTests(request: TestRunRequest): TestRunResult {
     // TODO unify with JavaTestRunner
-    val testCommand = System.getenv("EVALUATION_TEST_COMMAND")
+    val testCommand = request.testCommand
     requireNotNull(testCommand) { "Only test running via command is supported" }
 
     val testCommandParts = testCommand.split(" ")