Fixing enum case in output