Use enum instead of string