Since stochasiticity is naturally present in post-disruption plasma, and externally induced stochastization offers a prominent mechanism to mitigate RE avalanche, we present an advection-diffusion model that can be used to couple an orbit-following code to an orbit-averaged tool in order to capture the cross-field transport and to overcome the latter's limitation.