Change message to Message to match Go