Actually fix failing test