fix bug 

fix bug