diff --git a/docs/source/how-to/configuration/use-configuration-schema.ipynb b/docs/source/how-to/configuration/use-configuration-schema.ipynb
index 4869ab3..a46f34c 100644
--- a/docs/source/how-to/configuration/use-configuration-schema.ipynb
+++ b/docs/source/how-to/configuration/use-configuration-schema.ipynb
@@ -91,7 +91,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 14,
    "id": "0c2c9f7e",
    "metadata": {},
    "outputs": [
@@ -99,26 +99,19 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "pyaml_cs_oa.controlsystem.OphydAsyncControlSystem → OphydAsyncControlSystemConfigurationSchema\n",
-      "tango.pyaml.controlsystem.TangoControlSystem → TangoControlSystemConfigurationSchema\n"
+      "{'pyaml_cs_oa.controlsystem.OphydAsyncControlSystem': <class 'pyaml_cs_oa.controlsystem.OphydAsyncControlSystemConfigurationSchema'>,\n",
+      " 'tango.pyaml.controlsystem.TangoControlSystem': <class 'tango.pyaml.controlsystem.TangoControlSystemConfigurationSchema'>}\n"
      ]
     }
    ],
    "source": [
+    "from pprint import pprint\n",
+    "\n",
     "control_system_schema = registry[\"pyaml.control.controlsystem.ControlSystem\"]\n",
     "\n",
-    "control_system_options = {\n",
-    "    class_path: schema\n",
-    "    for class_path, schema in registry.items()\n",
-    "    if schema is not control_system_schema\n",
-    "    and (\n",
-    "        issubclass(schema, control_system_schema)\n",
-    "        or schema.is_virtual_subclass_of(control_system_schema)\n",
-    "    )\n",
-    "}\n",
+    "subclasses = registry.subclasses_of(control_system_schema)\n",
     "\n",
-    "for class_path, schema in sorted(control_system_options.items()):\n",
-    "    print(class_path, \"→\", schema.__name__)"
+    "pprint(subclasses)"
    ]
   },
   {