Now using the enum instead of a variable.