Fix argument error

- [x] Fix argument error