better proba handling