Fix breakage due to change of arg name