Fix return message format