diff --git a/tensorflow/tools/api/lib/python_object_to_proto_visitor.py b/tensorflow/tools/api/lib/python_object_to_proto_visitor.py
index aed4516b858..c7b754422a5 100644
--- a/tensorflow/tools/api/lib/python_object_to_proto_visitor.py
+++ b/tensorflow/tools/api/lib/python_object_to_proto_visitor.py
@@ -97,11 +97,17 @@ _NORMALIZE_TYPE[(
     'tensorflow.python.framework.ops.Tensor')] = (
         "<class 'tensorflow.python.framework.ops.Tensor'>")
 _NORMALIZE_TYPE['typing.Generic'] = "<class 'typing.Generic'>"
-# TODO(mdan): Remove once the golden files are generated in Python 3.7.
+# TODO(b/203104448): Remove once the golden files are generated in Python 3.7.
 _NORMALIZE_TYPE["<class 'typing._GenericAlias'>"] = 'typing.Union'
-# TODO(mdan): Remove once the golden files are generated in Python 3.9.
+# TODO(b/203104448): Remove once the golden files are generated in Python 3.9.
 _NORMALIZE_TYPE["<class 'typing._UnionGenericAlias'>"] = 'typing.Union'
-
+# TODO(b/203104448): Remove once the golden files are generated in Python 3.8.
+_NORMALIZE_TYPE[
+    "<class 'typing_extensions._ProtocolMeta'>"] = ("<class "
+                                                    "'typing._ProtocolMeta'>")
+# TODO(b/203104448): Remove once the golden files are generated in Python 3.8.
+_NORMALIZE_TYPE[
+    "<class 'typing_extensions.Protocol'>"] = "<class 'typing.Protocol'>"
 
 if sys.version_info.major == 3 and sys.version_info.minor >= 8:
   _NORMALIZE_TYPE["<class '_collections._tuplegetter'>"] = "<type 'property'>"