diff --git a/ipi-survey-platform/src/components/survey/ProductData/ProductData.jsx b/ipi-survey-platform/src/components/survey/ProductData/ProductData.jsx
index ae0c26a..d2e9d94 100644
--- a/ipi-survey-platform/src/components/survey/ProductData/ProductData.jsx
+++ b/ipi-survey-platform/src/components/survey/ProductData/ProductData.jsx
@@ -621,9 +621,8 @@ if (!establishment) {
       response = await submitSurvey(payload);
       
       // Update local storage with the new submission ID if this was a create operation
-      if (response && response.id) {
-        const updatedSubmission = { ...currentSubmission, id: response.id };
-        localStorage.setItem('currentSubmission', JSON.stringify(updatedSubmission));
+      if (response && response.submission_data) {
+        localStorage.setItem('currentSubmission', JSON.stringify(response.submission_data));
       }
       
       alert('Draft saved successfully')