[Extension] Fix DI Extension