diff --git a/intellij.tools.ide.starter/src/com/intellij/ide/starter/ide/IDETestContext.kt b/intellij.tools.ide.starter/src/com/intellij/ide/starter/ide/IDETestContext.kt
index a97b8f51..0c0e67a3 100644
--- a/intellij.tools.ide.starter/src/com/intellij/ide/starter/ide/IDETestContext.kt
+++ b/intellij.tools.ide.starter/src/com/intellij/ide/starter/ide/IDETestContext.kt
@@ -328,6 +328,8 @@ open class IDETestContext(
 
   fun enableHighlightingLog(logLevel: LogLevel = LogLevel.DEBUG): IDETestContext = applyVMOptionsPatch { configureLoggers(logLevel, "com.intellij.codeInsight") }
 
+  fun enableCheckTrafficLight(enable: Boolean = true): IDETestContext = applyVMOptionsPatch { addSystemProperty("is.test.traffic.light", enable) }
+
   fun removeIdeaProjectDirectory(): IDETestContext {
     val ideaDirPath = resolvedProjectHome.resolve(".idea")