Fix the bloody tests
2 files changed