Modified tests to match new page