fixes test breakages
Merge pull request #48 from xeno-by/master

fixes test breakages