Fix estimation bug
Merge pull request #9 from mattybeam/master

Fix estimation bug