diff --git a/src/code2graph/requirements.txt b/src/code2graph/requirements.txt
index e47236ba9..6841a0608 100644
--- a/src/code2graph/requirements.txt
+++ b/src/code2graph/requirements.txt
@@ -11,7 +11,7 @@ rdflib==4.2.2
 requests==2.21.0
 scikit-learn==0.20.2
 selenium==3.141.0
-tensorflow==1.15.0
+tensorflow==2.7.2
 urllib3==1.24.1
 wget==3.2
 lxml==4.3.4