another change to use the enum