small bugfix in DI