diff --git a/bec_widgets/utils/compact_popup.py b/bec_widgets/utils/compact_popup.py
index e4335861..8d4daef2 100644
--- a/bec_widgets/utils/compact_popup.py
+++ b/bec_widgets/utils/compact_popup.py
@@ -2,7 +2,6 @@ import time
 from types import SimpleNamespace
 
 from bec_qthemes import material_icon
-from PySide6.QtWidgets import QToolButton
 from qtpy.QtCore import Property, Qt, Signal
 from qtpy.QtGui import QColor
 from qtpy.QtWidgets import (
@@ -12,6 +11,7 @@ from qtpy.QtWidgets import (
     QPushButton,
     QSizePolicy,
     QSpacerItem,
+    QToolButton,
     QVBoxLayout,
     QWidget,
 )