Update to reflect arg change in script