勞動者生育或者施行計劃生育手術依法享受休假期間，用人單位應當支付其工資。勞動者因產前檢查和哺乳依法休假的，用人單位應當視同其正常勞動支付工資。

在休假完畢，公司無故辭退員工是需要按找法定情形給予補償的。如果有法定事由辭退的應該延長到哺乳期結束，如果沒有，應該按照《勞動合同法》第四十七條雙倍賠償。

法律依據：《勞動合同法》

第四十七條 經濟補償的計算

經濟補償按勞動者在本單位工作的年限，每滿一年支付一個月工資的標準向勞動者支付。六個月以上不滿一年的，按一年計算；不滿六個月的，向勞動者支付半個月工資的經濟補償。勞動者月工資高于用人單位所在直轄市、設區的市級人民政府公布的本地區上年度職工月平均工資三倍的，向其支付經濟補償的標準按職工月平均工資三倍的數額支付，向其支付經濟補償的年限最高不超過十二年。本條所稱月工資是指勞動者在勞動合同解除或者終止前十二個月的平均工資。