Chế độ thai sản trong trường hợp con chết sau sinh

Tôi có thai và nghỉ chế độ thai sản. Nhưng do tình hình xấu, con tôi vừa sinh ra thì chết (khoảng 3 phút). Khi đi làm lại, do chưa có giấy xác nhận của bệnh viện nên công ty tôi chỉ làm hồ sơ cho tôi được nghỉ 50 ngày do xác định đây là chết thai lưu. Tôi đã nghỉ hết 50 ngày. Sau đó, bệnh viện có giấy xác nhận là em bé chết sau khi sinh ra, không phải thai lưu. Theo quy định thì nếu em bé chết sau sinh, tôi sẽ được nghỉ 4 tháng. Vậy, với trường hợp này, tôi được nghỉ bao lâu?