diff --git a/sdcflows/utils/wrangler.py b/sdcflows/utils/wrangler.py
index 4ec346b372..c884c63519 100644
--- a/sdcflows/utils/wrangler.py
+++ b/sdcflows/utils/wrangler.py
@@ -291,6 +291,7 @@ def find_estimators(
     base_entities = {
         "subject": subject,
         "extension": [".nii", ".nii.gz"],
+        "part": ["mag", None],
         "scope": "raw",  # Ensure derivatives are not captured
     }