small DI fix.