fix button highlight