Fixed unittest2 incompatibility