fix unit test / code was ok