Fix compile error
Merge pull request #11 from dchest/patch-1

Fix compile error