fix test suite

Hi, 

Some minor changes to fix the test suite. Thanks for the gem!