Use unittest2 compatible import