fixed unittest bug