Update tests to match new copy