Fix build
Merge pull request #469 from bjz/fix-build

Fix build