so now I need to refactor it somehow to still be able to use the enums. Maybe with a simple if statement