Fix load tests