[minor] Added lazy loading function