bugs fix

fix