diff --git a/tools/testClient/TestCaseExecuteEngine.py b/tools/testClient/TestCaseExecuteEngine.py
index 522c52417ba..059e7556637 100644
--- a/tools/testClient/TestCaseExecuteEngine.py
+++ b/tools/testClient/TestCaseExecuteEngine.py
@@ -1,4 +1,8 @@
-import unittest
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
+
 from functools import partial
 import os
 import sys
@@ -37,4 +41,4 @@ class TestCaseExecuteEngine(object):
         
         unittest.TextTestRunner(stream=self.testResultLogFile, verbosity=2).run(suite)
         
-        
\ No newline at end of file
+        
diff --git a/tools/testClient/dbConnection.py b/tools/testClient/dbConnection.py
index f4748f3c909..e6135edf00f 100644
--- a/tools/testClient/dbConnection.py
+++ b/tools/testClient/dbConnection.py
@@ -2,6 +2,7 @@ import pymysql
 import cloudstackException
 import sys
 import os
+import traceback
 class dbConnection(object):
     def __init__(self, host="localhost", port=3306, user='cloud', passwd='cloud', db='cloud'):
         self.host = host
@@ -13,6 +14,7 @@ class dbConnection(object):
         try:
             self.db = pymysql.Connect(host=host, port=port, user=user, passwd=passwd, db=db)
         except:
+            traceback.print_exc()
             raise cloudstackException.InvalidParameterException(sys.exc_info())
         
     def __copy__(self):
@@ -73,4 +75,4 @@ if __name__ == "__main__":
     for i in range(10):
         result = db.execute("select job_status, created, last_updated from async_job where id=%d"%i)
         print result
-       
\ No newline at end of file
+