Fixed issue where save button would disappear