Fix buttons focus state