diff --git a/workshop_sections/wide_n_deep/wide_n_deep_flow2.ipynb b/workshop_sections/wide_n_deep/wide_n_deep_flow2.ipynb
index 05f7457..8c6f388 100644
--- a/workshop_sections/wide_n_deep/wide_n_deep_flow2.ipynb
+++ b/workshop_sections/wide_n_deep/wide_n_deep_flow2.ipynb
@@ -610,7 +610,7 @@
     "    \n",
     "export_folder = m.export_savedmodel(\n",
     "    export_dir_base = model_dir + '/export',\n",
-    "    input_fn=serving_input_fn\n",
+    "    serving_input_fn=serving_input_fn\n",
     ")\n",
     "\n",
     "print('model exported successfully to {}'.format(export_folder))"