Fixed unittest error.