unit test small update