issue #48 - fix broke unit tests