Небольшая доработка во избежание ошибок