Added missing unittest import.