Make tests work!

Closes #6