Mala poprawka w warunku
Poprawa dzialania zapisywania stanu