diff --git a/tensorflow/lite/python/op_hint.py b/tensorflow/lite/python/op_hint.py
index c7f49bdf4b6..159fcaa2bf3 100644
--- a/tensorflow/lite/python/op_hint.py
+++ b/tensorflow/lite/python/op_hint.py
@@ -1166,8 +1166,7 @@ def _get_correct_mapping(original_index, nodes):
     node_indices = nodes.keys()
     node_indices = sorted(node_indices)
     return node_indices[-1]
-  else:
-    return original_index
+  return original_index
 
 
 def _convert_op_hints_to_stubs_helper(