抗日战争胜利后，经中共中央营救，于1946年3月4日获释。5日即致电毛泽东和中共中央，要求实现他多年的愿望，加入中国共产党。中共中央于7日复电，称赞他忠诚地为中国民族解放与人民解放事业进行了20余年的奋斗，经历了种种严重的考验，决定接受他加入中国共产党。4月8日，由重庆赴延安出席全军整编会议，因途中飞机失事，在山西兴县黑茶山遇难。遗体葬于延安"四八"烈士陵园。遗著有《南昌暴动至潮汕的失败》、《现代战争的性质特点与指挥》等，他记录军旅生涯的部分照片，被收入《叶挺将军摄影集》。