1 files changed, 10 insertions, 5 deletions
diff --git a/gn3/db/__init__.py b/gn3/db/__init__.py
index 34a0236..9b1b36a 100644
--- a/gn3/db/__init__.py
+++ b/gn3/db/__init__.py
@@ -6,27 +6,32 @@ from typing_extensions import Protocol
 
 from gn3.db.metadata_audit import MetadataAudit
 from gn3.db.phenotypes import Phenotype
-from gn3.db.phenotypes import PublishXRef
+from gn3.db.phenotypes import Probeset
 from gn3.db.phenotypes import Publication
+from gn3.db.phenotypes import PublishXRef
+
 
 from gn3.db.metadata_audit import metadata_audit_mapping
 from gn3.db.phenotypes import phenotype_mapping
+from gn3.db.phenotypes import probeset_mapping
 from gn3.db.phenotypes import publication_mapping
 from gn3.db.phenotypes import publish_x_ref_mapping
 
 
 TABLEMAP = {
-    "metadata_audit": metadata_audit_mapping,
     "Phenotype": phenotype_mapping,
-    "PublishXRef": publish_x_ref_mapping,
+    "ProbeSet": probeset_mapping,
     "Publication": publication_mapping,
+    "PublishXRef": publish_x_ref_mapping,
+    "metadata_audit": metadata_audit_mapping,
 }
 
 DATACLASSMAP = {
-    "metadata_audit": MetadataAudit,
     "Phenotype": Phenotype,
-    "PublishXRef": PublishXRef,
+    "ProbeSet": Probeset,
     "Publication": Publication,
+    "PublishXRef": PublishXRef,
+    "metadata_audit": MetadataAudit,
 }