Fixed another failing test