Fix tests

Fixes #119