diff --git a/tests/unit_tests/test_tree_widget.py b/tests/unit_tests/test_tree_widget.py
index 657c3c58..c22ea8d0 100644
--- a/tests/unit_tests/test_tree_widget.py
+++ b/tests/unit_tests/test_tree_widget.py
@@ -1,5 +1,5 @@
 import pytest
-from PySide6.QtWidgets import QTreeView, QWidget
+from qtpy.QtWidgets import QTreeView, QWidget
 
 from bec_widgets.utils.colors import apply_theme