Fixed Message entity bug