Fix tests. Fixes #55.