diff --git a/src/regvelo/datasets/__init__.py b/src/regvelo/datasets/__init__.py
index 4567edc..f6d2d92 100644
--- a/src/regvelo/datasets/__init__.py
+++ b/src/regvelo/datasets/__init__.py
@@ -3,6 +3,7 @@
 from ._datasets import (
     zebrafish_nc,
     zebrafish_grn,
+    zebrafish_perturb,
     murine_nc,
     human_limb,
     hindbrain,
@@ -13,6 +14,7 @@
 __all__ = [
     "zebrafish_nc",
     "zebrafish_grn",
+    "zebrafish_perturb",
     "murine_nc",
     "human_limb",
     "hindbrain",