diff --git a/src/pystencils_autodiff/backends/_tensorflow.py b/src/pystencils_autodiff/backends/_tensorflow.py
index 58eb2c8daf4c332b395f3f82c8cbaf6721da6cde..4ecae0f84efb01ef86022f2aab6bc8ed986754f5 100644
--- a/src/pystencils_autodiff/backends/_tensorflow.py
+++ b/src/pystencils_autodiff/backends/_tensorflow.py
@@ -1,7 +1,7 @@
 import tensorflow as tf
 
 import pystencils_autodiff
-from tf.compat.v1 import get_default_graph
+from tensorflow.compat.v1 import get_default_graph
 
 _num_generated_ops = 0