diff --git a/python/minimal_test.py b/python/minimal_test.py
index 6924ace..3c0df48 100644
--- a/python/minimal_test.py
+++ b/python/minimal_test.py
@@ -29,16 +29,16 @@
 
 import unittest
 from google.protobuf.pyext import _message
-from google.protobuf.internal import api_implementation
+from google3.net.proto2.python.internal import api_implementation
 from google.protobuf import unittest_pb2
 from google.protobuf import map_unittest_pb2
 from google.protobuf import descriptor_pool
 from google.protobuf import text_format
 from google.protobuf import message_factory
 from google.protobuf import message
-from google.protobuf.internal import factory_test1_pb2
-from google.protobuf.internal import factory_test2_pb2
-from google.protobuf.internal import more_extensions_pb2
+from google3.net.proto2.python.internal import factory_test1_pb2
+from google3.net.proto2.python.internal import factory_test2_pb2
+from google3.net.proto2.python.internal import more_extensions_pb2
 from google.protobuf import descriptor_pb2
 
 class TestMessageExtension(unittest.TestCase):