Add check in each estimator function