Багфикс + исправление тестов