שעת הנקה לתינוק פג

שלום רב,
שאלתי היא בנושא שעת הנקה.
התינוק שלי נולד בחודש ה6 להריוני, היות כך הוא היה מאושפז שלושה חודשים. לאחר מכן חופשת הלידה שלי התארכה בעוד חודשיים ומיד לאחר מכן חזרתי לעבודה.
החוק היבש אומר כי שעת ההנקה מגיעה לאחר חופשת הלידה (3 חודשים) במשך 4 חודשים.
השאלה שלי היא: האם במצב הזה שהילד היה מאושפז מגיע לי כרגע שעת הנקה במשך חודשיים? או במשך 4 חודשים?
תודה על העזרה