Update to the gem