DI bugfixes