Fixed missing save call