Improve unit-tests.

Поправил unit - тесты