Add missing unittest2 import.