Fix bugs - Modify message return