Баг фикс

сори, забыл проверить