diff --git a/build-scripts/runtime_lib.cmake b/build-scripts/runtime_lib.cmake
index 461a64b37..3127fd66b 100644
--- a/build-scripts/runtime_lib.cmake
+++ b/build-scripts/runtime_lib.cmake
@@ -81,7 +81,8 @@ if (WAMR_BUILD_WASI_NN EQUAL 1)
                     RESULT_VARIABLE TENSORFLOW_RESULT
     )
     set(TENSORFLOW_SOURCE_DIR "/root/src/tflite/tensorflow_src")
-
+    include_directories (${CMAKE_CURRENT_BINARY_DIR}/flatbuffers/include)
+    include_directories (${TENSORFLOW_SOURCE_DIR})
     add_subdirectory(
         "${TENSORFLOW_SOURCE_DIR}/tensorflow/lite"
         "${CMAKE_CURRENT_BINARY_DIR}/tensorflow-lite" EXCLUDE_FROM_ALL)
diff --git a/core/iwasm/libraries/wasi-nn/wasi_nn.c b/core/iwasm/libraries/wasi-nn/wasi_nn.c
index 567080f67..422c9316c 100644
--- a/core/iwasm/libraries/wasi-nn/wasi_nn.c
+++ b/core/iwasm/libraries/wasi-nn/wasi_nn.c
@@ -25,7 +25,6 @@ uint32_t wasi_nn_load(wasm_exec_env_t exec_env, uint32_t builder, uint32_t encod
     printf("Inside wasi_nn_load!\n\n");
     wasm_module_inst_t instance = wasm_runtime_get_module_inst(exec_env);
     graph_builder_array buf = (graph_builder_array) wasm_runtime_addr_app_to_native(instance, builder);
-    load_return res;
     switch ((graph_encoding) encoding) {
         case openvino:
             return invalid_argument;