diff --git a/CMakeLists.txt b/CMakeLists.txt
index 8a5625d..caddb29 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -1,5 +1,5 @@
 cmake_minimum_required(VERSION 3.16)
-project(unitree_go2_custom VERSION 1.0.0)
+project(lzwc_go2 VERSION 1.0.0)
 
 # Set C++ standard
 set(CMAKE_CXX_STANDARD 17)
@@ -26,6 +26,8 @@ include_directories(
     $<$<NOT:$<BOOL:${PahoMqttC_FOUND}>>:${PAHO_MQTT_C_INCLUDE_DIR}>
 )
 
+target_include_directories(main PRIVATE ${unitree_sdk2_INCLUDE_DIRS})
+
 # Source files
 set(SOURCES
     src/main.cpp