; and (B) has been diagnosed with a substance use disorder. (2) Limitation; conditions (A) Limitation A State may only receive Federal financial participation for medical assistance provided to an individual described in paragraph (1) during the 7-day period that begins on the first day that the individual is an inmate of a public institution. (B) Conditions A State may only receive Federal financial participation for medical assistance provided to an individual described in paragraph (1) if— (i) the State has elected to not terminate eligibility for medical assistance under the State plan for individuals on the basis that they are inmates of public institutions (but may suspend coverage during the period an individual is such an inmate)