Fix null message bug