diff --git a/tfx/examples/chicago_taxi_pipeline/taxi_utils.py b/tfx/examples/chicago_taxi_pipeline/taxi_utils.py
index 2ca7167b22..214aa29de9 100644
--- a/tfx/examples/chicago_taxi_pipeline/taxi_utils.py
+++ b/tfx/examples/chicago_taxi_pipeline/taxi_utils.py
@@ -370,4 +370,4 @@ def run_fn(fn_args: fn_args_utils.FnArgs):
           model, tf_transform_output
       ),
   }
-  model.save(fn_args.serving_model_dir, save_format='tf', signatures=signatures)
+  tf.saved_model.save(model, fn_args.serving_model_dir, signatures=signatures)