Update test to match copy change