Небольшая правка

Поправил изменение легенды при нажатии кнопки undo