Fix entity error