fix bug in test.