Fixes to handle arg parameters