fix type hint.
Merge pull request #5532