Fix unittest erros