Fix the mute button issue