Fix DI Extension name