Keep selections after expansion