fix lazy load tests