Fixed button highlighting issue.