fix lazy loading not working