fix the arguments bug