diff --git a/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/settings/SettingsDialogUi.kt b/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/settings/SettingsDialogUi.kt
index 8908d1437e10..3e57a89dcf42 100644
--- a/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/settings/SettingsDialogUi.kt
+++ b/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/settings/SettingsDialogUi.kt
@@ -7,6 +7,7 @@ import com.intellij.driver.sdk.ui.components.UiComponent
 import com.intellij.driver.sdk.ui.components.common.IdeaFrameUI
 import com.intellij.driver.sdk.ui.components.common.WelcomeScreenUI
 import com.intellij.driver.sdk.ui.components.elements.DialogUiComponent
+import com.intellij.driver.sdk.ui.components.elements.JTreeUiComponent
 import com.intellij.driver.sdk.ui.components.elements.accessibleTree
 import com.intellij.driver.sdk.ui.should
 import javax.swing.JDialog
@@ -22,7 +23,7 @@ private fun Finder.onSettingsDialog(
   x(SettingsDialogUiComponent::class.java, locator).apply(action)
 
 open class SettingsDialogUiComponent(data: ComponentData) : DialogUiComponent(data) {
-  val settingsTree = accessibleTree { byAccessibleName("Settings categories") }
+  open val settingsTree: JTreeUiComponent = accessibleTree { byAccessibleName("Settings categories") }
 
   fun openTreeSettingsSection(vararg path: String, fullMatch: Boolean = true) {
     settingsTree.should(message = "Settings tree is empty", timeout = 5.seconds) { collectExpandedPaths().isNotEmpty() }