Fix build error.
Merge pull request #1 from xinfli/master

Fix build error.