diff --git a/mediapipe/calculators/tflite/tflite_tensors_to_classification_calculator.cc b/mediapipe/calculators/tflite/tflite_tensors_to_classification_calculator.cc
index 906b4242f..0ef4d5142 100644
--- a/mediapipe/calculators/tflite/tflite_tensors_to_classification_calculator.cc
+++ b/mediapipe/calculators/tflite/tflite_tensors_to_classification_calculator.cc
@@ -51,9 +51,9 @@ namespace mediapipe {
 //   input_stream: "TENSORS:tensors"
 //   output_stream: "CLASSIFICATIONS:classifications"
 //   options: {
-//     [mediapipe.TfLiteTensorsToClassificationCalculatorOptions.ext] {
-//       num_classes: 1024
+//     [type.googleapis.com/mediapipe.TfLiteTensorsToClassificationCalculatorOptions] {
 //       min_score_threshold: 0.1
+//       top_k: 5
 //       label_map_path: "labelmap.txt"
 //     }
 //   }