Could you use the enum name instead of the integer?