Fix unittest2 import