update unit test to last changes