diff --git a/lamindb_setup/_schema_metadata.py b/lamindb_setup/_schema_metadata.py
index 5e13f6e01..4c85aa204 100644
--- a/lamindb_setup/_schema_metadata.py
+++ b/lamindb_setup/_schema_metadata.py
@@ -120,7 +120,7 @@ def _get_modules_metadata(self):
                 for model in self._get_schema_module(
                     module_name
                 ).models.__dict__.values()
-                if model.__class__.__name__ == "ModelBase"
+                if model.__class__.__name__ == "RegistryMeta"
                 and model.__name__ not in ["Registry", "ORM"]
                 and not model._meta.abstract
                 and model.__get_schema_name__() == module_name