Fix enum problem