Fixed it so it actually works.

I should not release a gem without testing it first.