Fix build error
Merge pull request #18 from huei90/master

Fix build error