diff --git a/tosa_converter_for_tflite/tosa_converter_for_tflite_pybind11.cc b/tosa_converter_for_tflite/tosa_converter_for_tflite_pybind11.cc
index 1f40c34b674e579177477ee81bf8820f36893d70..22399b5fc0dec4f8651ba1b22c073dbf5d8a1143 100644
--- a/tosa_converter_for_tflite/tosa_converter_for_tflite_pybind11.cc
+++ b/tosa_converter_for_tflite/tosa_converter_for_tflite_pybind11.cc
@@ -116,10 +116,9 @@ PYBIND11_MODULE(_tosa_converter_for_tflite_wrapper, m) {
   m.def(
       "_tflite_flatbuffer_to_tosa_mlir",
       [](const std::string& tflite_input_file, pybind11::object tosa_sink,
+         tosa_converter_for_tflite::TosaConverterOutputFormat output_format,
          const std::unordered_map<std::string, std::vector<int64_t>>&
-             override_tflite_input_shape,
-         tosa_converter_for_tflite::TosaConverterOutputFormat output_format =
-             tosa_converter_for_tflite::TosaConverterOutputFormat::Bytecode) {
+             override_tflite_input_shape) {
         DispatchTFLiteToTosaConversion(tflite_input_file, tosa_sink,
                                        output_format,
                                        override_tflite_input_shape);
@@ -137,6 +136,6 @@ PYBIND11_MODULE(_tosa_converter_for_tflite_wrapper, m) {
             output_format (TosaConverterOutputFormat, optional):
                 - TosaConverterOutputFormat.Bytecode: emit MLIR bytecode (default)
                 - TosaConverterOutputFormat.Text: emit human-readable MLIR text
-            override_shapes (dict): Mapping of input tensor names to their overridden shape.
+            override_tflite_input_shape (dict): Mapping of input tensor names to their overridden shape.
       )doc");
 }