diff --git a/python/core/keyset_handle.py b/python/core/keyset_handle.py
index 8dcfd912ad58fb415d3802fd1902f2f1d322c87f..0d102ad62ce4538d02ab3e852333804ae8c8cd77 100644
--- a/python/core/keyset_handle.py
+++ b/python/core/keyset_handle.py
@@ -23,7 +23,7 @@ import random
 
 from typing import Type, TypeVar
 
-from google3.net.proto2.python.public import message
+from google.protobuf import message
 from tink.proto import tink_pb2
 from tink.python.aead import aead
 from tink.python.core import keyset_reader as reader
diff --git a/python/core/keyset_reader.py b/python/core/keyset_reader.py
index 22fab228512c58860108943938aae03d72d5fe5d..4d67ef132464dcdf40753db319c5b82bf00750ca 100644
--- a/python/core/keyset_reader.py
+++ b/python/core/keyset_reader.py
@@ -25,8 +25,8 @@ from typing import Text
 
 from tink.proto import tink_pb2
 from tink.python.core import tink_error
-from google3.net.proto2.python.public import json_format
-from google3.net.proto2.python.public import message
+from google.protobuf import json_format
+from google.protobuf import message
 
 
 class KeysetReader(object):
diff --git a/python/core/keyset_writer.py b/python/core/keyset_writer.py
index def31614e3b384af02ca5d2b013f436ad5790ab6..fcbefe5c53fa3d35f78289bfc9000d7bd01d3bd5 100644
--- a/python/core/keyset_writer.py
+++ b/python/core/keyset_writer.py
@@ -22,7 +22,7 @@ from __future__ import print_function
 import abc
 import io
 
-from google3.net.proto2.python.public import json_format
+from google.protobuf import json_format
 from tink.proto import tink_pb2
 from tink.python.core import tink_error