1 files changed, 2 insertions, 4 deletions
diff --git a/lldb/test/tools/lldb-mi/TestMiPrompt.py b/lldb/test/tools/lldb-mi/TestMiPrompt.py
index 32ae53281e3..6f4bb76ddc7 100644
--- a/lldb/test/tools/lldb-mi/TestMiPrompt.py
+++ b/lldb/test/tools/lldb-mi/TestMiPrompt.py
@@ -2,9 +2,10 @@
 Test that the lldb-mi driver prints prompt properly.
 """
 
+import lldb_shared
+
 import lldbmi_testcase
 from lldbtest import *
-import unittest2
 
 class MiPromptTestCase(lldbmi_testcase.MiTestCaseBase):
 
@@ -50,6 +51,3 @@ class MiPromptTestCase(lldbmi_testcase.MiTestCaseBase):
         # Test that lldb-mi is ready after program exited
         self.expect("\*stopped,reason=\"exited-normally\"")
         self.expect(self.child_prompt, exactly = True)
-
-if __name__ == '__main__':
-    unittest2.main()