BUG: small bug in estimate.py