Replace gem to add enum attribute support