Updated entity enum