diff --git a/docs/tutorials/bert_glue.ipynb b/docs/tutorials/bert_glue.ipynb
index 5497e8eda..30a2fc6d0 100644
--- a/docs/tutorials/bert_glue.ipynb
+++ b/docs/tutorials/bert_glue.ipynb
@@ -227,7 +227,7 @@
         "  tf.tpu.experimental.initialize_tpu_system(cluster_resolver)\n",
         "  strategy = tf.distribute.TPUStrategy(cluster_resolver)\n",
         "  print('Using TPU')\n",
-        "elif tf.test.is_gpu_available():\n",
+        "elif tf.config.list_physical_devices('GPU'):\n",
         "  strategy = tf.distribute.MirroredStrategy()\n",
         "  print('Using GPU')\n",
         "else:\n",