OSAWATOMIE, Kan. (AP) — Federal authorities are no longer threatening to end Medicare and Medicaid funding for the Osawatomie State Hospital after work began on $3 million worth safety improvements.

The Kansas City Star reports that the Department of Health and Human Services told the psychiatric facility Monday that it had made sufficient progress and payments that amount to about one-fourth of the hospital’s $26 million annual budget would continue.

Osawatomie was threatened with the loss of funding after the department inspected the facility in January and found that it was not in compliance with several safety standards to prevent patient injuries and suicides.

The hospital has been replacing beds, installing new bathroom fixtures and replacing suspended ceilings. They have also been eliminating items that could be used for hanging or as a weapon.