fix lazy loading in tests