update to follow original gem