diff --git a/lldb/packages/Python/lldbsuite/test/lldbplaygroundrepl.py b/lldb/packages/Python/lldbsuite/test/lldbplaygroundrepl.py
index 58626f9ff5d29..f336530628681 100644
--- a/lldb/packages/Python/lldbsuite/test/lldbplaygroundrepl.py
+++ b/lldb/packages/Python/lldbsuite/test/lldbplaygroundrepl.py
@@ -120,4 +120,4 @@ def test_playgrounds(self):
     import atexit
     lldb.SBDebugger.Initialize()
     atexit.register(lldb.SBDebugger.Terminate)
-    unittest2.main()
+    unittest.main()
diff --git a/lldb/test/API/python_api/sbvalue_updates/TestSBValueUpdates.py b/lldb/test/API/python_api/sbvalue_updates/TestSBValueUpdates.py
index 4dadfc84b512b..f54f73bd04843 100644
--- a/lldb/test/API/python_api/sbvalue_updates/TestSBValueUpdates.py
+++ b/lldb/test/API/python_api/sbvalue_updates/TestSBValueUpdates.py
@@ -66,4 +66,4 @@ def do_test(self):
     import atexit
     lldb.SBDebugger.Initialize()
     atexit.register(lldb.SBDebugger.Terminate)
-    unittest2.main()
+    unittest.main()