diff --git a/generate_model_code.sh b/generate_model_code.sh
index 3310efd833e424b3a4b2543bb5fca84fbff4dbb0..d893e29f4997547c2c966c3a3a5cc9325352a77c 100644
--- a/generate_model_code.sh
+++ b/generate_model_code.sh
@@ -20,7 +20,7 @@ bazel-bin/lib/python/tools/converter --platform=${PLATFORM} \
                                      --data_type=${DATA_TYPE} \
                                      --runtime=${RUNTIME} \
                                      --output_type=source \
-                                     --template=${LIBMACE_SOURCE_DIR}/lib/python/tools/model.template \
+                                     --template=${LIBMACE_SOURCE_DIR}/lib/python/tools \
                                      --model_tag=${MODEL_TAG} \
                                      --input_shape=${INPUT_SHAPES} \
                                      ${DSP_MODE_FLAG} \