Fix function args type

[Goralt]

Resolves #NUMBER

Description:
A very high-level summary of easily-reproducible changes that can be understood by non-devs.

Type of change:

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to change)

Why is this change required?:
A simple explanation of what the problem is and how this PR solves it

Code changes:

• A bullet point list of key code changes that have been made.
• When describing code changes, try to communicate how and why you implemented something a specific way, not just what has changed.

---

Checklist:

• My code follows the code style of this project.
• My change requires a change to the documentation.
• I have updated the documentation accordingly.
• I have read the CONTRIBUTING document.
• I have added tests to cover my changes.
• All new and existing tests passed.