Fix save message display toggle