Make all training algorithms work with changes