fixed bug with button