Fix build error
Merge pull request #3 from marksamman/master

Fix build error