Маленькое добавление

Убрал небольшой баг.