Removes strata column when not needed