Исправил ожидание в тесте.