update as per changes to gem