Fix the tests failing.