fixed a couple more bugs in the gem