Fixed register selection after saving.