Fix bug in for_brownian_motion