fix type hint.
Merge pull request #5532 from zhanxu/patch-1

fix type hint.