Follow-up fix :)