diff --git a/.github/workflows/integration-tests.yaml b/.github/workflows/integration-tests.yaml
index 1f9d5b8f9d..4eb63428bc 100644
--- a/.github/workflows/integration-tests.yaml
+++ b/.github/workflows/integration-tests.yaml
@@ -55,6 +55,22 @@ jobs:
             python-version: "3.10"
 
     steps:
+      # This step is a Workaround to avoid the "No space left on device" error.
+      # ref: https://github.com/actions/runner-images/issues/2840
+      - name: Remove unnecessary files
+        shell: bash
+        run: |
+          sudo rm -rf /usr/share/dotnet
+          sudo rm -rf /opt/ghc
+          sudo rm -rf "/usr/local/share/boost"
+          sudo rm -rf "$AGENT_TOOLSDIRECTORY"
+          sudo rm -rf /usr/local/lib/android
+          sudo rm -rf /usr/local/share/powershell
+          sudo rm -rf /usr/share/swift
+          
+          echo "Disk usage after cleanup:"
+          df -h
+
       - name: Checkout
         uses: actions/checkout@v3
 
diff --git a/sdk/python/test/e2e/test_e2e_xgboostjob.py b/sdk/python/test/e2e/test_e2e_xgboostjob.py
index 1c334e2b78..9b8dc5e2bd 100644
--- a/sdk/python/test/e2e/test_e2e_xgboostjob.py
+++ b/sdk/python/test/e2e/test_e2e_xgboostjob.py
@@ -181,7 +181,7 @@ def generate_xgboostjob(
 def generate_container() -> V1Container:
     return V1Container(
         name=CONTAINER_NAME,
-        image="docker.io/merlintang/xgboost-dist-iris:1.1",
+        image="docker.io/kubeflow/xgboost-dist-iris:latest",
         args=[
             "--job_type=Train",
             "--xgboost_parameter=objective:multi:softprob,num_class:3",