Fix argument size issue