Fix up some tests.
2 files changed