Updates tests to use new script