A State to which a grant is made under section
603 of this title and which has a State plan approved under subchapter XIX of this chapter shall provide that in the case of a family that is treated (under section
1396u–1
(b)(1)(A) of this title for purposes of subchapter XIX of this chapter) as receiving aid under a State plan approved under this part (as in effect on July 16, 1996), that would become ineligible for such aid because of hours of or income from employment of the caretaker relative (as defined under this part as in effect on such date) or because of section
602
(a)(8)(B)(ii)(II) of this title (as so in effect), and that was so treated as receiving such aid in at least 3 of the 6 months immediately preceding the month in which such ineligibility begins, the family shall remain eligible for medical assistance under the State’s plan approved under subchapter XIX of this chapter for an extended period or periods as provided in section
1396r–6 or
1396a
(e)(1) of this title (as applicable), and that the family will be appropriately notified of such extension as required by section
1396r–6
(a)(2) of this title.