Enum bug fix