correct the initial distribution of p