; and (II) the availability of medication-assisted treatment and medically supervised withdrawal management services for such individuals. (ii) Required completion The State compete an assessment described in clause (i) not later than 12 months after the date the State commences such assessment. . (3) Clarification of levels of care Section 1915(l)(7)(A) of the Social Security Act ( 42 U.S.C. 1396n(l)(7)(A) ) is amended by inserting (or any successor publication) before the period. 305. Prohibition on termination of enrollment due to incarceration (a) Medicaid (1) In general Section 1902(a)(84)(A) of the Social Security Act ( 42 U.S.C. 1396a(a)(86)(A) ), as amended by section 5122(a)(2) of the Consolidated Appropriations Act, 2023 ( Public Law 117–328 ), is further amended— (A) by striking under the State plan and inserting under the State plan (or waiver of such plan) ; (B) by striking who is an eligible juvenile (as defined in subsection (nn)(2)) ; (C) by striking because the juvenile and inserting because the individual ; (D) by striking during the period the juvenile and inserting during the period the individual ; and (E) by inserting such an individual who is an eligible juvenile (as defined in subsection (nn)(2)) or a woman during pregnancy (and during the 60-day beginning on the last day of pregnancy) and after or in the case of . (2) Effective date The amendments made by— (A) subparagraph (A) of paragraph (1) shall take effect on the date of the enactment of this Act; and (B) subparagraphs (B) through (E) of paragraph (1) shall take effect on January 1, 2025. (b) CHIP (1) In general Section 2102(d)(1)(A) of the Social Security Act ( 42 U.S.C. 1397bb(d)(1)(A) ) is amended— (A) by inserting or pregnancy-related after child health ; (B) by inserting or targeted low-income pregnant woman after targeted low-income child ; (C) by inserting or pregnant woman after because the child