diff --git a/src/features/grid/api/save-draft-changes.ts b/src/features/grid/api/save-draft-changes.ts
index 86a4aca..3456661 100644
--- a/src/features/grid/api/save-draft-changes.ts
+++ b/src/features/grid/api/save-draft-changes.ts
@@ -5,6 +5,8 @@ import { ventesProduits } from "@/db/schema";
 import { eq, and } from "drizzle-orm";
 import { z } from "zod";
 
+import { revalidatePath } from "next/cache";
+
 const SaveDraftsSchema = z.object({
     changes: z.array(
         z.object({
@@ -26,21 +28,23 @@ export async function saveDraftChanges(
     const { changes } = parsed.data;
 
     try {
-        // Batch update all rows for given codein × magasin
+        // Batch update all rows for given codein. 
+        // If magasin is "TOTAL", update ALL stores for that product.
         await Promise.all(
-            changes.map(({ codein, magasin, codeGamme }) =>
-                db
+            changes.map(({ codein, magasin, codeGamme }) => {
+                const whereClause = [eq(ventesProduits.codein, codein)];
+                if (magasin !== "TOTAL") {
+                    whereClause.push(eq(ventesProduits.magasin, magasin));
+                }
+
+                return db
                     .update(ventesProduits)
                     .set({ codeGamme })
-                    .where(
-                        and(
-                            eq(ventesProduits.codein, codein),
-                            eq(ventesProduits.magasin, magasin)
-                        )
-                    )
-            )
+                    .where(and(...whereClause));
+            })
         );
 
+        revalidatePath("/grid");
         return { success: true, saved: changes.length };
     } catch (err) {
         const msg = err instanceof Error ? err.message : "Unknown error";
diff --git a/src/features/grid/store/use-grid-store.ts b/src/features/grid/store/use-grid-store.ts
index 6af1745..9647e4c 100644
--- a/src/features/grid/store/use-grid-store.ts
+++ b/src/features/grid/store/use-grid-store.ts
@@ -82,6 +82,7 @@ export const useGridStore = create<GridState>()(
             partialize: (state) => ({
                 filters: state.filters,
                 displayDensity: state.displayDensity,
+                draftChanges: state.draftChanges,
             }),
         }
     )