Fix existing test
Merge pull request #2 from pickl-it/fix-test

Fix existing test