diff --git a/samples/core/parameterized_tfx_oss/taxi_pipeline_notebook.ipynb b/samples/core/parameterized_tfx_oss/taxi_pipeline_notebook.ipynb
index b138be86023..92e56459ca9 100644
--- a/samples/core/parameterized_tfx_oss/taxi_pipeline_notebook.ipynb
+++ b/samples/core/parameterized_tfx_oss/taxi_pipeline_notebook.ipynb
@@ -239,7 +239,7 @@
     "      trainer, model_analyzer, model_validator, pusher\n",
     "  ],\n",
     "  enable_cache=True,\n",
-    "  beam_pipeline_args=['--direct_num_workers=%d' % 4],\n",
+    "  beam_pipeline_args=['--direct_num_workers=%d' % 0],\n",
     ")"
    ]
   },
@@ -311,4 +311,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
\ No newline at end of file
+}