Switched to enum