We don't need a enum. But the class-name and module attributes were changed/added.