Update to changes in enum