(i) Requires the State to prepare a corrective action plan or improvement plan if the Secretary determines that the State should be able to correct the problem within one year.