Fixed broken test
Merge pull request #130 from iceglow/cleanup

Fixed broken test