Fixed an issue with button behavior.