Fix this to use an Integer field instead, Enum doesn't work