Fix bug for di configuration