diff --git a/tensorflow/lite/micro/examples/hello_world/evaluate.py b/tensorflow/lite/micro/examples/hello_world/evaluate.py
index 0c252fdc71f..f6f7ac1b62d 100644
--- a/tensorflow/lite/micro/examples/hello_world/evaluate.py
+++ b/tensorflow/lite/micro/examples/hello_world/evaluate.py
@@ -108,7 +108,8 @@ def get_tflite_prediction(model_path, x_values):
 
   for i, x_value in enumerate(x_values):
     y_predictions[i] = invoke_tflite_interpreter(
-        in_interpreter  tflite_interpreter,
+        input_shape,
+        tflite_interpreter,
         x_value,
         input_details['index'],
         output_details['index'],