Fixed issue with button toggling.