Fix to make it work with DI extensions in neon