fix bug for testing