diff --git a/tools/shell/CMakeLists.txt b/tools/shell/CMakeLists.txt
index f62a5b2204..342d6410d2 100644
--- a/tools/shell/CMakeLists.txt
+++ b/tools/shell/CMakeLists.txt
@@ -26,9 +26,9 @@ ELSE ()
     SET(LINK_WEBSOCKET "")
 ENDIF ()
 
-IF (CUS_NAME OR CUS_PROMPT)
+IF (CUS_NAME OR CUS_PROMPT OR CUS_EMAIL)
     ADD_DEFINITIONS(-I${CMAKE_CURRENT_SOURCE_DIR}/../../../enterprise/packaging)
-ENDIF (CUS_NAME)
+ENDIF (CUS_NAME OR CUS_PROMPT OR CUS_EMAIL)
 
 if(TD_WINDOWS)
     target_link_libraries(shell PUBLIC taos_static ${LINK_WEBSOCKET})