Add lazy load fix