Небольшая модификация для теста.