1 files changed, 2 insertions, 4 deletions
diff --git a/lldb/test/tools/lldb-mi/signal/TestMiSignal.py b/lldb/test/tools/lldb-mi/signal/TestMiSignal.py
index 16532c8d13e..22f05c31952 100644
--- a/lldb/test/tools/lldb-mi/signal/TestMiSignal.py
+++ b/lldb/test/tools/lldb-mi/signal/TestMiSignal.py
@@ -2,9 +2,10 @@
 Test that the lldb-mi handles signals properly.
 """
 
+import lldb_shared
+
 import lldbmi_testcase
 from lldbtest import *
-import unittest2
 
 class MiSignalTestCase(lldbmi_testcase.MiTestCaseBase):
 
@@ -197,6 +198,3 @@ class MiSignalTestCase(lldbmi_testcase.MiTestCaseBase):
         # Exit
         self.runCmd("-gdb-exit")
         self.expect("\^exit")
-
-if __name__ == '__main__':
-    unittest2.main()