fix bug for test