fixing failing unit tests

fixes #20