Trying to fix Enum