Use enum instead of number