Fix tests after changes introduced by #86

A little adjustment to make the tests pass again. It was annoying to not see the "build pass" green flag on the README ;)