diff --git a/.github/workflows/e2e-testing.yml b/.github/workflows/e2e-testing.yml
index d64ceab3..c096a44e 100644
--- a/.github/workflows/e2e-testing.yml
+++ b/.github/workflows/e2e-testing.yml
@@ -27,6 +27,9 @@ jobs:
       
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
+        with:
+          driver-opts: |
+            network=host
       
       - name: Make start-all.sh executable
         run: chmod +x testinit/start-all.sh testinit/stop-all.sh
diff --git a/libazureinit/Cargo.toml b/libazureinit/Cargo.toml
index 6188839d..eb4bb4f2 100644
--- a/libazureinit/Cargo.toml
+++ b/libazureinit/Cargo.toml
@@ -28,7 +28,7 @@ tokio-util = "0.7"
 sysinfo = "0.38"
 anyhow = "1"
 fstab = "0.4.0"
-toml = "0.9"
+toml = "1.0"
 regex = "1"
 lazy_static = "1.4"
 figment = { version = "0.10", features = ["toml"] }
diff --git a/testinit/Dockerfile b/testinit/Dockerfile
index 4bc3d67c..46cf7ad5 100644
--- a/testinit/Dockerfile
+++ b/testinit/Dockerfile
@@ -8,7 +8,8 @@ VOLUME [ "/sys/fs/cgroup" ]
 
 RUN if command -v apt-get >/dev/null 2>&1; then \
     # Debian/Ubuntu (apt-based) \
-    apt-get update && apt-get install -y \
+    apt-get -o Acquire::Retries=3 update \
+    && apt-get -o Acquire::Retries=3 install -y \
     systemd \
     dbus \
     curl \