Fix the build error
Merge pull request #2753 from Randy-Ma/fix_build_error

Fix the build error