fix bug 2

fix bug 2