Import unittest2 if its needed