diff --git a/docs/notebooks/import_catalogs.ipynb b/docs/notebooks/import_catalogs.ipynb
index be48c224..d5d9f47d 100644
--- a/docs/notebooks/import_catalogs.ipynb
+++ b/docs/notebooks/import_catalogs.ipynb
@@ -114,7 +114,8 @@
     "    catalog_name=catalog_from_dataframe,\n",
     "    catalog_type=\"object\",\n",
     "    highest_order=5,\n",
-    "    threshold=100\n",
+    "    threshold=100,\n",
+    "    append_partition_info=True\n",
     ")\n",
     "\n",
     "# Save it to disk in HiPSCat format\n",
@@ -281,7 +282,7 @@
     "# Verify that pixels are similar\n",
     "assert from_dataframe_catalog.get_healpix_pixels() == from_importer_catalog.get_healpix_pixels()\n",
     "# Verify that resulting dataframes contain the same data\n",
-    "# pd.testing.assert_frame_equal(from_dataframe_catalog._ddf.compute(), from_importer_catalog._ddf.compute())"
+    "pd.testing.assert_frame_equal(from_dataframe_catalog.compute().sort_index(), from_importer_catalog.compute().sort_index(), check_dtype=False)"
    ]
   }
  ],