AE: Correct the x' distribution by removing some stuff that I didn't need