Adding code to import IInjectionContext