Fix test build
Merge pull request #4 from ebfe/fix-build

Fix test build