Fix an insufficiency when saving eigenstates