made tests work!

Imagine that :)