diff --git a/flaml/default/estimator.py b/flaml/default/estimator.py
index 717e3e195..ca7c980da 100644
--- a/flaml/default/estimator.py
+++ b/flaml/default/estimator.py
@@ -39,8 +39,8 @@ def flamlize_estimator(super_class, name: str, task: str, alternatives=None):
             self._params = params
             super().__init__(**params)
 
-        @wraps(super_class._get_param_names)
         @classmethod
+        @wraps(super_class._get_param_names)
         def _get_param_names(cls):
             return super_class._get_param_names()