Added missing save call