button bug fix

Signed-off-by: peskish <exodus3412@gmail.com>