Fix a bug with an incorrect parameter type

[gengxinwu]

Describe your changes

Fix a bug with an incorrect parameter type.

Checklist before requesting a review

• Add unit tests for this change.
• Make sure all tests can pass.
• Update documents if necessary.
• Lint and apply fixes to your code by running lintrunner -a
• Is this a user-facing change? If yes, give a description of this change to be included in the release notes.

(Optional) Issue link

[gengxinwu]

please help merge if it looks good, thanks @jambayk