diff --git a/crossfit/op/base.py b/crossfit/op/base.py
index 528046c..e544785 100644
--- a/crossfit/op/base.py
+++ b/crossfit/op/base.py
@@ -70,7 +70,7 @@ def add_keep_cols(self, data, output):
                 output[col] = data[col]
 
         columns = list(output.columns)
-        # we use dict.fromkeys to remove duplicates and preserve order
+        # we use dict.fromkeys to remove duplicates and preserve order (to match _build_dask_meta behavior) 
         output = output[list(dict.fromkeys(self.keep_cols + columns))]
 
         return output