Fixing tests.
2 files changed