fix bug with button display