diff --git a/plugin/src/main/kotlin/com/github/grishberg/profiler/ui/PluginThemeController.kt b/plugin/src/main/kotlin/com/github/grishberg/profiler/ui/PluginThemeController.kt
index 5a8e75f..9bd5372 100644
--- a/plugin/src/main/kotlin/com/github/grishberg/profiler/ui/PluginThemeController.kt
+++ b/plugin/src/main/kotlin/com/github/grishberg/profiler/ui/PluginThemeController.kt
@@ -4,11 +4,11 @@ import com.github.grishberg.profiler.chart.theme.DarkPalette
 import com.github.grishberg.profiler.chart.theme.LightPalette
 import com.github.grishberg.profiler.ui.theme.Palette
 import com.github.grishberg.profiler.ui.theme.ThemeController
-import com.intellij.util.ui.UIUtil
+import com.intellij.util.ui.StartupUiUtil.isDarkTheme
 import javax.swing.JMenuBar
 
 class PluginThemeController : ThemeController {
-    private val dark = UIUtil.isUnderDarcula()
+    private val dark = isDarkTheme;
 
     override val palette: Palette = if (dark) DarkPalette() else LightPalette()