Minor changes to build the gem