fixes #92 - fix tests