Updates to correction and sampling