Fix a bug about function argument.