Fix broken test
Merge pull request #130 from mmun/fix-test

Fix broken test