better behavior of the unit test