Use the enum instead of int?