diff --git a/bec_widgets/utils/container_utils.py b/bec_widgets/utils/container_utils.py
index 02bac4dd..9b4875d5 100644
--- a/bec_widgets/utils/container_utils.py
+++ b/bec_widgets/utils/container_utils.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import itertools
 from typing import Type
 
diff --git a/bec_widgets/widgets/bec_queue/bec_queue.py b/bec_widgets/widgets/bec_queue/bec_queue.py
index 04dfd7d7..a9575ce6 100644
--- a/bec_widgets/widgets/bec_queue/bec_queue.py
+++ b/bec_widgets/widgets/bec_queue/bec_queue.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from bec_lib.endpoints import MessageEndpoints
 from qtpy.QtCore import Qt, Slot
 from qtpy.QtWidgets import QHBoxLayout, QHeaderView, QTableWidget, QTableWidgetItem, QWidget
diff --git a/docs/conf.py b/docs/conf.py
index 7401ae3e..7cb22231 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -65,7 +65,7 @@ add_module_names = False  # Remove namespaces from class/method signatures
 autodoc_inherit_docstrings = True  # If no docstring, inherit from base class
 set_type_checking_flag = True  # Enable 'expensive' imports for sphinx_autodoc_typehints
 autoclass_content = "both"  # Include both class docstring and __init__
-autodoc_mock_imports = ["pyqtgraph"]
+autodoc_mock_imports = ["pyqtgraph", "qtpy", "PySide6"]
 
 # Add any paths that contain templates here, relative to this directory.
 templates_path = ["_templates"]