Fix LazyLoading error