Fix buga przy edycji