Fix the import alias for unittest2