Fix unittest import.