Poprawienie błędu po zakończeniu treningu

Poprawiony błąd występujący po zakończeniu treningu.