diff --git a/compute_kernel_writer/prototype/include/ckw/OperandBase.h b/compute_kernel_writer/prototype/include/ckw/OperandBase.h
index a9e313f..06d9f82 100644
--- a/compute_kernel_writer/prototype/include/ckw/OperandBase.h
+++ b/compute_kernel_writer/prototype/include/ckw/OperandBase.h
@@ -25,7 +25,7 @@
 #ifndef CKW_PROTOTYPE_INCLUDE_CKW_OPERANDBASE_H
 #define CKW_PROTOTYPE_INCLUDE_CKW_OPERANDBASE_H
 
-#include "ckw/Types.h"
+#include "ckw/types/DataType.h"
 #include <string>
 
 namespace ckw