Merge pull request #438 from YorkShen/bugfix/build_error

Fix build error