fix a small button issue