diff --git a/advanced_functionality/tensorflow_iris_byom/tensorflow_BYOM_iris.ipynb b/advanced_functionality/tensorflow_iris_byom/tensorflow_BYOM_iris.ipynb
index 40e09575d3..9e41db17b1 100644
--- a/advanced_functionality/tensorflow_iris_byom/tensorflow_BYOM_iris.ipynb
+++ b/advanced_functionality/tensorflow_iris_byom/tensorflow_BYOM_iris.ipynb
@@ -276,7 +276,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "classifier.save(\"export/Servo/1\")\n",
+    "classifier.export("export/Servo/1", format="tf_saved_model")\n",
     "with tarfile.open(\"model.tar.gz\", \"w:gz\") as tar:\n",
     "    tar.add(\"export\")"
    ]
@@ -469,4 +469,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
\ No newline at end of file
+}