When you rebake your maps after reimporting a corrected mesh the colors of the ID map changes. This cause all the layers that used color selection to break.
If the pick color remembered where you did your pick color it could update easily all the colors for the masks.