updating to correct for a DI change