Poprawa funkcji ;-)