diff --git a/validate_model.sh b/validate_model.sh
index 2c7f59a4a2dd96bcca1482e306da6c46c2fd9688..1cfa30d10c3fb20b4614ebdca42b0aa82426ee07 100755
--- a/validate_model.sh
+++ b/validate_model.sh
@@ -143,9 +143,9 @@ bazel-bin/lib/python/tools/tf_converter --input=${TF_MODEL_FILE_PATH} \
                                         --template=${LIBMACE_SOURCE_DIR}/lib/python/tools/model.template \
                                         --model_tag=${MODEL_TAG} \
                                         ${DSP_MODE_FLAG} \
+                                        --input_shape="${INPUT_SHAPE}" \
                                         --obfuscate=True || exit -1
 
-
 echo "Step 3: Download mace static library"
 download_and_link_lib