fixed tests

Hey Scott,

As discussed yesterday I figured out how to get the tests right :)

cheers :beers: