Fix argument logic.