diff --git a/code/preproc/src/setup.py b/code/preproc/src/setup.py
index 62eca17..208eb02 100644
--- a/code/preproc/src/setup.py
+++ b/code/preproc/src/setup.py
@@ -8,7 +8,7 @@
 REQUIRED_PACKAGES = [
     'apache-beam[gcp]==2.11.0',
     'tensorflow-transform==0.13.0',
-    'tensorflow==1.13.1'
+    'tensorflow==2.11.1'
 ]
 
 setuptools.setup(