Improve the cache strategy of DI