diff --git a/trackastra/data/data.py b/trackastra/data/data.py
index 6bf04a2..620dc36 100644
--- a/trackastra/data/data.py
+++ b/trackastra/data/data.py
@@ -124,7 +124,7 @@ def __init__(
             "patch",
             "patch_regionprops",
             "wrfeat",
-        ] = "none",
+        ] = "wrfeat",
         sanity_dist: bool = False,
         crop_size: tuple | None = None,
         return_dense: bool = False,