Updated DI usage in dibi vs ndbt