fixing broken test from merging pull request #8.