Fix typo
This fixes some of the tests failing
Improved 33071d1
Merge pull request #441 from iulianR/master

Small change to fix most of the tests