diff --git a/bec_widgets/widgets/dap_combo_box/dap_combo_box.py b/bec_widgets/widgets/dap_combo_box/dap_combo_box.py
index 57ea05dd..2ff31fb9 100644
--- a/bec_widgets/widgets/dap_combo_box/dap_combo_box.py
+++ b/bec_widgets/widgets/dap_combo_box/dap_combo_box.py
@@ -165,21 +165,18 @@ class DapComboBox(BECWidget, QWidget):
         return True
 
 
-# pragma: no cover
-def main():
-    """Main function to run the DapComboBox widget."""
-    import sys
-
+if __name__ == "__main__":  # pragma: no cover
+    # pylint: disable=import-outside-toplevel
     from qtpy.QtWidgets import QApplication
 
     from bec_widgets.utils.colors import set_theme
 
-    app = QApplication(sys.argv)
-    set_theme("auto")
-    widget = DapComboBox()
+    app = QApplication([])
+    set_theme("dark")
+    widget = QWidget()
+    widget.setFixedSize(200, 200)
+    layout = QVBoxLayout()
+    widget.setLayout(layout)
+    layout.addWidget(DapComboBox())
     widget.show()
-    sys.exit(app.exec_())
-
-
-if __name__ == "__main__":
-    main()
+    app.exec_()
diff --git a/docs/assets/widget_screenshots/dap_combo_box.png b/docs/assets/widget_screenshots/dap_combo_box.png
index b5565a2e..1fa315e6 100644
Binary files a/docs/assets/widget_screenshots/dap_combo_box.png and b/docs/assets/widget_screenshots/dap_combo_box.png differ