Fix another regression