bugs fix

bugs fix