багфикс бакгфикса

Сделал немного по другому, более правильно ))