add missing lazy_load import