diff --git a/rider-fsharp/src/test/kotlin/com/jetbrains/rider/plugins/fsharp/test/cases/FSharpCompletionTest.kt b/rider-fsharp/src/test/kotlin/com/jetbrains/rider/plugins/fsharp/test/cases/FSharpCompletionTest.kt
index 0017f245b7..843c4f82b2 100644
--- a/rider-fsharp/src/test/kotlin/com/jetbrains/rider/plugins/fsharp/test/cases/FSharpCompletionTest.kt
+++ b/rider-fsharp/src/test/kotlin/com/jetbrains/rider/plugins/fsharp/test/cases/FSharpCompletionTest.kt
@@ -5,6 +5,7 @@ import com.jetbrains.rider.test.annotations.TestSettings
 import com.jetbrains.rider.test.base.PatchEngineCompletionTestBase
 import com.jetbrains.rider.test.enums.BuildTool
 import com.jetbrains.rider.test.enums.sdk.SdkVersion
+import com.jetbrains.rider.test.facades.editor.PatchEngineEditorTestMode
 import com.jetbrains.rider.test.scriptingApi.dumpOpenedEditorFacade
 import org.testng.annotations.Test
 
diff --git a/rider-fsharp/src/test/kotlin/com/jetbrains/rider/plugins/fsharp/test/cases/typingAssist/FSharpPatchEngineTypingAssists.kt b/rider-fsharp/src/test/kotlin/com/jetbrains/rider/plugins/fsharp/test/cases/typingAssist/FSharpPatchEngineTypingAssists.kt
index f61d162b36..b9a2063eff 100644
--- a/rider-fsharp/src/test/kotlin/com/jetbrains/rider/plugins/fsharp/test/cases/typingAssist/FSharpPatchEngineTypingAssists.kt
+++ b/rider-fsharp/src/test/kotlin/com/jetbrains/rider/plugins/fsharp/test/cases/typingAssist/FSharpPatchEngineTypingAssists.kt
@@ -7,6 +7,7 @@ import com.jetbrains.rider.test.annotations.TestSettings
 import com.jetbrains.rider.test.base.PatchEngineEditorTestBase
 import com.jetbrains.rider.test.enums.BuildTool
 import com.jetbrains.rider.test.enums.sdk.SdkVersion
+import com.jetbrains.rider.test.facades.editor.PatchEngineEditorTestMode
 import com.jetbrains.rider.test.logging.TestLoggerHelper
 import com.jetbrains.rider.test.reporting.SubsystemConstants
 import com.jetbrains.rider.test.scriptingApi.changeFileContent
diff --git a/rider-fsharp/src/test/resources/intellij.rider.plugins.fsharp.test.cases.xml b/rider-fsharp/src/test/resources/intellij.rider.plugins.fsharp.test.cases.xml
index 41e5c37f6c..1f2d14885b 100644
--- a/rider-fsharp/src/test/resources/intellij.rider.plugins.fsharp.test.cases.xml
+++ b/rider-fsharp/src/test/resources/intellij.rider.plugins.fsharp.test.cases.xml
@@ -2,7 +2,8 @@
     <dependencies>
         <plugin id="com.jetbrains.rider.fsharp"/>
         <module name="intellij.rider.test.framework"/>
-        <module name="intellij.rider.test.framework._test"/>
+        <module name="intellij.rider.test.framework.integration.testng"/>
+        <module name="intellij.rider.test.framework.integration.testng._test"/>
         <module name="intellij.rider.test.framework.core"/>
         <module name="intellij.rider.test.framework.testng"/>
         <module name="intellij.rider.test.framework.junit"/>