Fix tests

They were not updated to match 95a862b49a1864a4e32c39fab5cca029574d4026
Merge pull request #4 from noidi/fix-tests

Fix tests