Made the gem work with the right initializers