Fix test errors.
2 files changed