diff --git a/CMakeLists.txt b/CMakeLists.txt
index e501a54..c51d0af 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -120,7 +120,6 @@
 
 # Include directories for application module:
 set(APPLICATION_INCLUDE_DIRS
-    ${SRC_PATH}/application/tensorflow-lite-micro/include
     ${SRC_PATH}/application/main/include)
 
 # Source files for application module:
diff --git a/scripts/cmake/cmsis-pack-gen/CMakeLists.txt b/scripts/cmake/cmsis-pack-gen/CMakeLists.txt
index 2774ab2..19787b6 100644
--- a/scripts/cmake/cmsis-pack-gen/CMakeLists.txt
+++ b/scripts/cmake/cmsis-pack-gen/CMakeLists.txt
@@ -55,6 +55,7 @@
 # Any custom steps that are required for the CMSIS pack generation flow
 # must be declared here:
 if (CMSIS_PACK_GEN_FLOW)
+    target_include_directories(common_api PUBLIC ${TENSORFLOW_SRC_PATH})
     # For CMSIS packs, we need CMSIS DSP definition.
     # @TODO: Currently, this line is added multiple times to the include file. Uncomment when
     # packgen is fixed.