Fix tests

Fixes #53.