Get rid of unittest2 dependency