fix work with button. not test(!).