Correctly importing unittest2