diff --git a/dags/solutions_team/solutionsteam_tf_nightly_supported.py b/dags/solutions_team/solutionsteam_tf_nightly_supported.py
index c2bebeed..d79bfdf1 100644
--- a/dags/solutions_team/solutionsteam_tf_nightly_supported.py
+++ b/dags/solutions_team/solutionsteam_tf_nightly_supported.py
@@ -41,6 +41,7 @@
       tpu_zone=Zone.US_CENTRAL1_C.value,
       time_out_in_min=60,
       global_batch_size=1024,
+      runtime_version=RuntimeVersion.V2_ALPHA_TPUV5.value,
   )
 
   tf_resnet_v3_8 = tf_config.get_tf_resnet_config(
@@ -48,6 +49,7 @@
       tpu_cores=8,
       tpu_zone=Zone.US_EAST1_D.value,
       time_out_in_min=60,
+      runtime_version=RuntimeVersion.V2_ALPHA_TPUV5.value,
   )
 
   tf_resnet_v4_8 = tf_config.get_tf_resnet_config(
@@ -55,6 +57,7 @@
       tpu_cores=8,
       tpu_zone=Zone.US_CENTRAL2_B.value,
       time_out_in_min=60,
+      runtime_version=RuntimeVersion.V2_ALPHA_TPUV5.value,
   )
 
   tf_resnet_v4_32 = tf_config.get_tf_resnet_config(