Fix argument parser