diff --git a/python/google/protobuf/internal/well_known_types_test.py b/python/google/protobuf/internal/well_known_types_test.py
index 60b0c47d03..0e31e6f807 100644
--- a/python/google/protobuf/internal/well_known_types_test.py
+++ b/python/google/protobuf/internal/well_known_types_test.py
@@ -36,10 +36,14 @@ __author__ = 'jieluo@google.com (Jie Luo)'
 
 from datetime import datetime
 
+try:
+  import unittest2 as unittest
+except ImportError:
+  import unittest
+
 from google.protobuf import duration_pb2
 from google.protobuf import field_mask_pb2
 from google.protobuf import timestamp_pb2
-import unittest
 from google.protobuf import unittest_pb2
 from google.protobuf.internal import test_util
 from google.protobuf.internal import well_known_types