Fix argument type error