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

Fix the build error