diff --git a/taskcluster/tc-cpp_tflite-tests-prod.sh b/taskcluster/tc-cpp_tflite-tests-prod.sh
index 23a0d8a7..2d0b969a 100644
--- a/taskcluster/tc-cpp_tflite-tests-prod.sh
+++ b/taskcluster/tc-cpp_tflite-tests-prod.sh
@@ -2,6 +2,10 @@
 
 set -xe
 
+#TODO: Remove after 0.6.1
+export DEEPSPEECH_PROD_MODEL=https://github.com/lissyx/DeepSpeech/releases/download/v0.6.0/output_graph.tflite
+export DEEPSPEECH_PROD_MODEL_MMAP=https://github.com/lissyx/DeepSpeech/releases/download/v0.6.0/output_graph.tflite
+
 source $(dirname "$0")/tc-tests-utils.sh
 
 model_source=${DEEPSPEECH_PROD_MODEL//.pb/.tflite}
diff --git a/taskcluster/tc-node_tflite-tests-prod.sh b/taskcluster/tc-node_tflite-tests-prod.sh
index 55fb05ad..aef3b48e 100644
--- a/taskcluster/tc-node_tflite-tests-prod.sh
+++ b/taskcluster/tc-node_tflite-tests-prod.sh
@@ -2,6 +2,10 @@
 
 set -xe
 
+#TODO: Remove after 0.6.1
+export DEEPSPEECH_PROD_MODEL=https://github.com/lissyx/DeepSpeech/releases/download/v0.6.0/output_graph.tflite
+export DEEPSPEECH_PROD_MODEL_MMAP=https://github.com/lissyx/DeepSpeech/releases/download/v0.6.0/output_graph.tflite
+
 source $(dirname "$0")/tc-tests-utils.sh
 
 nodever=$1
@@ -32,4 +36,4 @@ npm install --prefix ${NODE_ROOT} --cache ${NODE_CACHE} ${deepspeech_npm_url}
 
 check_runtime_nodejs
 
-run_prod_inference_tests
+run_prodtflite_inference_tests
diff --git a/taskcluster/tc-python_tflite-tests-prod.sh b/taskcluster/tc-python_tflite-tests-prod.sh
index abae7cf8..6b6ae765 100644
--- a/taskcluster/tc-python_tflite-tests-prod.sh
+++ b/taskcluster/tc-python_tflite-tests-prod.sh
@@ -2,6 +2,10 @@
 
 set -xe
 
+#TODO: Remove after 0.6.1
+export DEEPSPEECH_PROD_MODEL=https://github.com/lissyx/DeepSpeech/releases/download/v0.6.0/output_graph.tflite
+export DEEPSPEECH_PROD_MODEL_MMAP=https://github.com/lissyx/DeepSpeech/releases/download/v0.6.0/output_graph.tflite
+
 source $(dirname "$0")/tc-tests-utils.sh
 
 extract_python_versions "$1" "pyver" "pyver_pkg" "py_unicode_type" "pyconf" "pyalias"
@@ -46,6 +50,6 @@ LD_LIBRARY_PATH=${PY37_LDPATH}:$LD_LIBRARY_PATH pip install --verbose --only-bin
 which deepspeech
 deepspeech --version
 
-run_prod_inference_tests
+run_prodtflite_inference_tests
 
 virtualenv_deactivate "${pyalias}" "${PYENV_NAME}"