Fix build failure
Merge pull request #3 from dlackty/patch-1

Fix build failure