Make enum correct