fix buttom bug