bugfix: actually take into accout parameters for get_probability