diff --git a/README.md b/README.md
index f0af061..15f3149 100644
--- a/README.md
+++ b/README.md
@@ -32,6 +32,7 @@ To build the pointpillars inference, **TensorRT** with PillarScatter layer and *
 $ cd test
 $ mkdir build
 $ cd build
+$ cmake .
 $ make -j$(nproc)
 ```
 
diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
index 373a696..cf14d82 100644
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -16,6 +16,12 @@
 cmake_minimum_required(VERSION 2.8.7)
 set(PROJECT_NAME demo)
 find_package(CUDA REQUIRED)
+include_directories(
+    ${CUDA_INCLUDE_DIRS}
+    ${TENSORRT_INCLUDE_DIRS}
+    ../include/
+    ../include/plugin/
+)
 
 set(CMAKE_C_COMPILER /usr/bin/aarch64-linux-gnu-gcc)
 set(CMAKE_CXX_COMPILER /usr/bin/aarch64-linux-gnu-g++)
@@ -49,13 +55,6 @@ set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS}
 set(TENSORRT_INCLUDE_DIRS /usr/include/aarch64-linux-gnu/)
 set(TENSORRT_LIBRARY_DIRS /usr/lib/aarch64-linux-gnu/)
 
-include_directories(
-    ${CUDA_INCLUDE_DIRS}
-    ${TENSORRT_INCLUDE_DIRS}
-    ../include/
-    ../include/plugin/
-)
-
 link_directories(
 	${TENSORRT_LIBRARY_DIRS}
 	/usr/lib/aarch64-linux-gnu