Update test to not depend on pagination