FIX DI issue