fix to get back enum classes working