diff --git a/tsk/ui/button.py b/tsk/ui/button.py
index 59bef756..ebd9c613 100644
--- a/tsk/ui/button.py
+++ b/tsk/ui/button.py
@@ -4,7 +4,7 @@ from typing import Callable, List, Dict, Any
 import pyray as rl
 
 from openpilot.system.ui.lib.application import gui_app
-from openpilot.system.ui.lib.button import gui_button
+from openpilot.system.ui.widgets.button import gui_button
 from tsk.ui.layout import Theme  # Import Theme
 
 
diff --git a/tsk/ui/dialog.py b/tsk/ui/dialog.py
index c7a2c4e0..ad531268 100644
--- a/tsk/ui/dialog.py
+++ b/tsk/ui/dialog.py
@@ -2,8 +2,8 @@
 import pyray as rl
 
 from openpilot.system.ui.lib.application import gui_app
-from openpilot.system.ui.lib.button import gui_button, DEFAULT_BUTTON_FONT_SIZE
 from openpilot.system.ui.lib.scroll_panel import GuiScrollPanel
+from openpilot.system.ui.widgets.button import gui_button, DEFAULT_BUTTON_FONT_SIZE
 
 
 class BaseDialog: