diff --git a/distribution_shift_framework/requirements.txt b/distribution_shift_framework/requirements.txt
index 46c3735..2bc8653 100644
--- a/distribution_shift_framework/requirements.txt
+++ b/distribution_shift_framework/requirements.txt
@@ -49,7 +49,7 @@ tabulate==0.8.9
 tensorboard==2.8.0
 tensorboard-data-server==0.6.1
 tensorboard-plugin-wit==1.8.1
-tensorflow==2.8.0
+tensorflow==2.12.1
 tensorflow-datasets==4.5.2
 tensorflow-io-gcs-filesystem==0.24.0
 tensorflow-metadata==1.6.0