fixed unittest code.