Fix argument parsing error