diff --git a/dags/solutions_team/configs/tensorflow/common.py b/dags/solutions_team/configs/tensorflow/common.py
index 1c7628f4..92849820 100644
--- a/dags/solutions_team/configs/tensorflow/common.py
+++ b/dags/solutions_team/configs/tensorflow/common.py
@@ -120,7 +120,8 @@ def set_up_keras(version: Optional[str] = None) -> tuple[str]:
 
   return (
       cmd_install_keras,
-      "export PATH=$PATH:/root/google-cloud-sdk/bin && cd /tmp && sudo gcloud source repos clone tf2-api-tests --project=cloud-ml-auto-solutions",
+      "export PATH=$PATH:/root/google-cloud-sdk/bin && cd /tmp",
+      "gcloud source repos clone tf2-api-tests --project=cloud-ml-auto-solutions || (cd tf2-api-tests && git pull)"
       "cd /tmp/tf2-api-tests && pip install behave matplotlib",
   )