Changed 'entity' to 'gem' to fix a bug