diff --git a/taskcluster/tc-lite_benchmark_model-ds-tests.sh b/taskcluster/tc-lite_benchmark_model-ds-tests.sh
index 580e4a1d..662e2162 100755
--- a/taskcluster/tc-lite_benchmark_model-ds-tests.sh
+++ b/taskcluster/tc-lite_benchmark_model-ds-tests.sh
@@ -14,7 +14,7 @@ export PATH=${TASKCLUSTER_TMP_DIR}/ds/:$PATH
 lite_benchmark_model \
 	--graph=${TASKCLUSTER_TMP_DIR}/ds/${model_name} \
 	--show_flops \
-	--input_layer=input_node,previous_state_c,previous_state_h \
-	--input_layer_type=float,float,float \
-	--input_layer_shape=1,16,19,26:1:1,494:1,494 \
-	--output_layer=logits,new_state_c,new_state_h
+	--input_layer=input_node,previous_state_c,previous_state_h,input_samples \
+	--input_layer_type=float,float,float,float \
+	--input_layer_shape=1,16,19,26:1,100:1,100:512 \
+	--output_layer=logits,new_state_c,new_state_h,mfccs