fix issue in estimator.py