Fix unit test expectations.