Fix typo

Fixes #387
Fix problems with tests