Highlight button effect fix