fix bug in test 2