Fix compilation error.
Merge pull request #11 from danbeam/compile-fix

Fix compilation error.