diff --git a/vfxClientToolkit/ui/combo_checkbox.py b/vfxClientToolkit/ui/combo_checkbox.py
index 89742e7..dcb4573 100644
--- a/vfxClientToolkit/ui/combo_checkbox.py
+++ b/vfxClientToolkit/ui/combo_checkbox.py
@@ -1,4 +1,4 @@
-from qtpy.QtWidgets import QComboBox, QItemDelegate
+from qtpy.QtWidgets import QComboBox, QItemDelegate, QCheckBox
 from qtpy.QtGui import QStandardItem, QStandardItemModel
 
 class ComboCheckBoxDelegate(QItemDelegate):