Can you refactor this to be an enum?