Небольшой багфикс и тест.