diff --git a/medcat/compare_models/model_comparison.ipynb b/medcat/compare_models/model_comparison.ipynb
index c9eb022..2f5b4b4 100644
--- a/medcat/compare_models/model_comparison.ipynb
+++ b/medcat/compare_models/model_comparison.ipynb
@@ -127,7 +127,11 @@
     "from compare import get_diffs_for\n",
     "from output import parse_and_show, show_dict_deep, compare_dicts\n",
     "\n",
-    "cdb_comp, tally1, tally2, ann_diffs = get_diffs_for(model_path_1, model_path_2, documents_file)"
+    "cui_filter = None\n",
+    "# optional CUI filter:\n",
+    "# cui_filter = {\"25064002\"}\n",
+    "\n",
+    "cdb_comp, tally1, tally2, ann_diffs = get_diffs_for(model_path_1, model_path_2, documents_file, cui_filter=cui_filter)"
    ]
   },
   {