Set message when entity saved

* Set message when entity saved