fix dwell_prob to work with new sde class