diff --git a/R/seed.R b/R/seed.R
index cdc9322c4..456c8d6bd 100644
--- a/R/seed.R
+++ b/R/seed.R
@@ -10,7 +10,9 @@ tensorflow_on_before_use_session <- function(quiet) {
 }
 
 tensorflow_on_use_session <- function(sess, quiet) {
-  if (is_backend("tensorflow"))
-    keras$backend$set_session(sess)
+  if (is_backend("tensorflow")) {
+    if (tensorflow::tf_version() < "2.0")
+      keras$backend$set_session(sess)
+  }
 }