Fix failing tests.
2 files changed