Исправление

Доработка условия.