update for test 2 fix bug