Fix the unit test assertions.