Fix argument passing error