Fix a couple more tests.
2 files changed