Fix plot update problems