diff --git a/modules/bar/components/tray/TrayItem.qml b/modules/bar/components/tray/TrayItem.qml
index e199d8d..1e05ff7 100644
--- a/modules/bar/components/tray/TrayItem.qml
+++ b/modules/bar/components/tray/TrayItem.qml
@@ -1,20 +1,20 @@
 pragma ComponentBehavior: Bound
 
-import qs.widgets
+import qs.components
 import QtQuick
 import Quickshell
 import Quickshell.Services.SystemTray
 import Quickshell.Widgets
 import "menu/"
 
-StyledButton {
+StyledIconButton {
     id: root
 
     property SystemTrayItem trayItem
 
     onClicked: menu.toggle()
 
-    content: IconImage {
+    contentItem: IconImage {
         id: icon
         asynchronous: true
         implicitSize: 18