Небольшой рефакторинг, починил поломанные тесты