diff --git a/kokoro/build.sh b/kokoro/build.sh
index d48802d5..c6140eca 100644
--- a/kokoro/build.sh
+++ b/kokoro/build.sh
@@ -21,8 +21,10 @@ set -e
 cd "${KOKORO_ARTIFACTS_DIR}/github/dataflux-pytorch"
 
 function setup_virtual_envs() {
+    sudo apt-get -y update
+
     echo Setting up Python virtual environment.
-    sudo apt install python3.8-venv
+    sudo apt install -y python3.8-venv
     python3 -m venv venv
     source venv/bin/activate
 }
diff --git a/kokoro/presubmit.sh b/kokoro/presubmit.sh
index b5241cef..0c5b2dbf 100644
--- a/kokoro/presubmit.sh
+++ b/kokoro/presubmit.sh
@@ -21,8 +21,10 @@ set -e
 cd "${KOKORO_ARTIFACTS_DIR}/github/dataflux-pytorch"
 
 function setup_virtual_envs() {
+    sudo apt-get -y update
+
     echo Setting up Python virtual environment.
-    sudo apt install python3.8-venv
+    sudo apt install -y python3.8-venv
     python3 -m venv venv
     source venv/bin/activate
 }