Fixed test
Merge pull request #443 from ppp0/issue-443

Test not reflecting change in code