Небольшой фикс в коде