Fix flash messages. Set $I_entity protected to allow usage from children class