Fix Estimator mock