diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 1ed8833a..8a508128 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -13,7 +13,7 @@ repos:
     - repo: https://github.com/astral-sh/ruff-pre-commit
       rev: v0.12.1
       hooks:
-          - id: ruff
+          - id: ruff-check
             args: [--fix, --exit-non-zero-on-fix, --unsafe-fixes]
           - id: ruff-format
     - repo: https://github.com/pre-commit/pre-commit-hooks
diff --git a/docs/tutorials/notebooks b/docs/tutorials/notebooks
index 80914ce2..f7accc3c 160000
--- a/docs/tutorials/notebooks
+++ b/docs/tutorials/notebooks
@@ -1 +1 @@
-Subproject commit 80914ce243b16be9da1bb605831c8465e034df73
+Subproject commit f7accc3ca01d81d4408cf8c57e4f740365a0c514
diff --git a/pertpy/tools/_perturbation_space/_simple.py b/pertpy/tools/_perturbation_space/_simple.py
index 7c76b241..aa7de81b 100644
--- a/pertpy/tools/_perturbation_space/_simple.py
+++ b/pertpy/tools/_perturbation_space/_simple.py
@@ -169,10 +169,11 @@ def compute(
 
         if mode in ps_adata.layers:
             ps_adata.X = ps_adata.layers[mode]
+
         for col in original_obs.columns:
             if col not in ps_adata.obs.columns:
                 grouped_values = original_obs.groupby(grouping_cols)[col].first()
-                ps_adata.obs[col] = grouped_values.values
+                ps_adata.obs[col] = grouped_values.reindex(ps_adata.obs.index).values
 
         ps_adata.obs[target_col] = ps_adata.obs[target_col].astype("category")