diff --git a/script/UnitTest.py b/script/UnitTest.py
index 5c4071d..28db36f 100644
--- a/script/UnitTest.py
+++ b/script/UnitTest.py
@@ -1,7 +1,7 @@
 import ut as unittest
-reload(unittest)
+reload(ut)
 
-class IntegerArithmenticTestCase(unittest.TestCase):
+class Test(ut.TestCase):
     def testAdd(self):  ## test method names begin 'test*'
         self.assertEquals((1 + 2), 3)
         self.assertEquals(0 + 1, 1)
@@ -10,6 +10,7 @@ class IntegerArithmenticTestCase(unittest.TestCase):
         self.assertEquals((5 * 8), 40)
 
 if __name__ == '__main__':
-    unittest.main()
+    import sys;sys.argv = ['', 'Test.testName']
+    ut.main()
 
         
\ No newline at end of file