Fix enum bug