bugfix: was computing a prob for