minor fix with buttons