Energy conservation and saturation in small-x evolution