fix broken code after rust changes on enums