diff --git a/notebooks/worldcereal_v1_demo_custom_croptype.ipynb b/notebooks/worldcereal_v1_demo_custom_croptype.ipynb
index 39c5a89..c5ae4df 100644
--- a/notebooks/worldcereal_v1_demo_custom_croptype.ipynb
+++ b/notebooks/worldcereal_v1_demo_custom_croptype.ipynb
@@ -146,7 +146,7 @@
     "processing_period = slider.get_processing_period()\n",
     "\n",
     "# Query our public database of training data\n",
-    "public_df = query_public_extractions(polygon, processing_period=processing_period)\n",
+    "public_df = query_public_extractions(polygon, processing_period=processing_period, buffer=250000)\n",
     "public_df.year.value_counts()"
    ]
   },
diff --git a/notebooks/worldcereal_v1_demo_custom_croptype_extended.ipynb b/notebooks/worldcereal_v1_demo_custom_croptype_extended.ipynb
index 2a9f55c..e948c65 100644
--- a/notebooks/worldcereal_v1_demo_custom_croptype_extended.ipynb
+++ b/notebooks/worldcereal_v1_demo_custom_croptype_extended.ipynb
@@ -159,7 +159,7 @@
     "processing_period = slider.get_processing_period()\n",
     "\n",
     "# Query our public database of training data\n",
-    "public_df = query_public_extractions(polygon, processing_period=processing_period)\n",
+    "public_df = query_public_extractions(polygon, processing_period=processing_period, buffer=250000)\n",
     "public_df.year.value_counts()"
    ]
   },