单位需要赔偿其两个月工资。职工在工伤期间，被非法辞退的，可以要求其按照经济补偿金的2倍支付赔偿金。
职工在单位工作满一年的，单位应当补偿其两个月的工资；半年以上不满一年的，应当补偿两个月的工资；不满半年的，补偿一个月工资。
根据《劳动法》第29条:
有下列情形之一的，用人单位不得依照本法第二十六条、第二十七条的规定解除劳动合同:
(一)患职业病或者因工负伤并被确认丧失或者部分丧失劳动能力的；
(二)患病或者负伤，在规定的医疗期内的；
(三)女职工在孕期、产期、哺乳期的；
(四)法律、行政法规规定的其他情形。
根据《劳动合同法》第四十七条:
经济补偿金按照劳动者在本单位的工作年限支付，每满一年支付一个月的工资。
六个月以上不满一年的，按一年计算；不满六个月的，向劳动者支付半个月的工资。