Improve unit-tests.

Поправил unit - тесты
зачем?