update to use enum