fixes #196 - fix tests