diff --git a/src/brush/estimator.py b/src/brush/estimator.py
index 9cf68b4f..d076dd2c 100644
--- a/src/brush/estimator.py
+++ b/src/brush/estimator.py
@@ -155,6 +155,7 @@ def __init__(
         self.max_gen=max_gen
         self.verbosity=verbosity
         self.algorithm=algorithm
+        self.pick_criteria=pick_criteria
         self.mode=mode
         self.max_depth=max_depth
         self.max_size=max_size