Fixing test suite to work as gem