BOSTON (AP) — Massachusetts will begin limiting how long homeless families can stay in shelters as the state continues to grapple with an influx of homeless migrants.

Beginning June 1, the total length of stay will be limited to nine months, at the end of which families will be eligible for up to two 90-day extensions, under a supplemental budget approved by state lawmakers and signed by Democratic Gov. Maura Healey on Tuesday.

The law also requires the state Executive Office of Housing and Livable Communities to create a rehousing plan and provide case management for all individuals in shelters after they leave the program and requires the plan to be recertified every 60 days so that families can remain eligible.

Extensions would be based on circumstances that include employment or participation in a workforce training program, veteran status, imminent placement in housing, avoiding educational interruptions for children in public school and pregnancy or having recently given birth.