Fix tests
Merge pull request #17 from prometheus/fix-test

Fix tests