diff --git a/deepin-draw/widgets/ccheckbutton.h b/deepin-draw/widgets/ccheckbutton.h
index 76da835..fd7537a 100644
--- a/deepin-draw/widgets/ccheckbutton.h
+++ b/deepin-draw/widgets/ccheckbutton.h
@@ -21,6 +21,7 @@
 
 #include <QMap>
 #include <DToolButton>
+#include <DWidget>
 
 DWIDGET_USE_NAMESPACE
 
diff --git a/deepin-draw/widgets/cclickbutton.h b/deepin-draw/widgets/cclickbutton.h
index 7da95b4..d1f0eed 100644
--- a/deepin-draw/widgets/cclickbutton.h
+++ b/deepin-draw/widgets/cclickbutton.h
@@ -20,6 +20,7 @@
 #define CCLICKBUTTON_H
 
 #include <DToolButton>
+#include <DWidget>
 
 DWIDGET_USE_NAMESPACE