diff --git a/integration_test/third_party_apps_test/applications/dcgm/centos_rhel/install b/integration_test/third_party_apps_test/applications/dcgm/centos_rhel/install
index b6c65a18e2..9831b116f8 100644
--- a/integration_test/third_party_apps_test/applications/dcgm/centos_rhel/install
+++ b/integration_test/third_party_apps_test/applications/dcgm/centos_rhel/install
@@ -51,6 +51,7 @@ setup_repo() {
 install_cuda_from_package_manager() {
     setup_repo
     install_driver_package
+    # TODO(b/377558109): remove the temporary fix once the repo is updated
     sudo yum -y install cuda-toolkit cuda-demo* 
     verify_driver
 }
diff --git a/integration_test/third_party_apps_test/applications/dcgm/debian_ubuntu/install b/integration_test/third_party_apps_test/applications/dcgm/debian_ubuntu/install
index 170ab4b480..563ef22a13 100644
--- a/integration_test/third_party_apps_test/applications/dcgm/debian_ubuntu/install
+++ b/integration_test/third_party_apps_test/applications/dcgm/debian_ubuntu/install
@@ -34,6 +34,7 @@ case $DEVICE_CODE in
         ;;
     *)
         # For newer GPUs, install the latest version
+        # TODO(b/377558109): remove the temporary fix once the repo is updated
         sudo apt -y install cuda-12-6
         ;;
 esac
diff --git a/integration_test/third_party_apps_test/applications/dcgmv1/centos_rhel/install b/integration_test/third_party_apps_test/applications/dcgmv1/centos_rhel/install
index d1c1263f25..674ef516da 100644
--- a/integration_test/third_party_apps_test/applications/dcgmv1/centos_rhel/install
+++ b/integration_test/third_party_apps_test/applications/dcgmv1/centos_rhel/install
@@ -51,6 +51,7 @@ setup_repo() {
 install_cuda_from_package_manager() {
     setup_repo
     install_driver_package
+    # TODO(b/377558109): remove the temporary fix once the repo is updated
     sudo yum -y install cuda-toolkit cuda-demo*
     verify_driver
 }
diff --git a/integration_test/third_party_apps_test/applications/dcgmv1/debian_ubuntu/install b/integration_test/third_party_apps_test/applications/dcgmv1/debian_ubuntu/install
index 170ab4b480..563ef22a13 100644
--- a/integration_test/third_party_apps_test/applications/dcgmv1/debian_ubuntu/install
+++ b/integration_test/third_party_apps_test/applications/dcgmv1/debian_ubuntu/install
@@ -34,6 +34,7 @@ case $DEVICE_CODE in
         ;;
     *)
         # For newer GPUs, install the latest version
+        # TODO(b/377558109): remove the temporary fix once the repo is updated
         sudo apt -y install cuda-12-6
         ;;
 esac
diff --git a/integration_test/third_party_apps_test/applications/nvml/centos_rhel/install b/integration_test/third_party_apps_test/applications/nvml/centos_rhel/install
index 91352ff9b3..bc30b4c599 100644
--- a/integration_test/third_party_apps_test/applications/nvml/centos_rhel/install
+++ b/integration_test/third_party_apps_test/applications/nvml/centos_rhel/install
@@ -51,6 +51,7 @@ setup_repo() {
 install_cuda_from_package_manager() {
     setup_repo
     install_driver_package
+    # TODO(b/377558109): remove the temporary fix once the repo is updated
     sudo yum -y install cuda-toolkit cuda-demo*
     verify_driver
 }
diff --git a/integration_test/third_party_apps_test/applications/nvml/debian_ubuntu/install b/integration_test/third_party_apps_test/applications/nvml/debian_ubuntu/install
index 5c3cdf85f6..a4dfd08bd5 100644
--- a/integration_test/third_party_apps_test/applications/nvml/debian_ubuntu/install
+++ b/integration_test/third_party_apps_test/applications/nvml/debian_ubuntu/install
@@ -34,6 +34,7 @@ case $DEVICE_CODE in
         ;;
     *)
         # For newer GPUs, install the latest version
+        # TODO(b/377558109): remove the temporary fix once the repo is updated 
         sudo apt -y install cuda-12-6
         ;;
 esac