Updates tests to use codeception