fix

Poprawa buga.