Shouldn't you use `cls` instead of `EntityRole` here?