Switch to unittest2