Fix for setting the DI container at CLI