diff --git a/Tools/storage_migrate.sh b/Tools/storage_migrate.sh
index cbb86f6..83adb9e 100755
--- a/Tools/storage_migrate.sh
+++ b/Tools/storage_migrate.sh
@@ -203,20 +203,24 @@ else
 fi
 
 # ──────────────────────────────────────────────────────────────────────────────
-# Step 4: Update TARGET_DIR in master.conf (new location)
+# Step 4: Update TARGET_DIR, DATA_DIR, STATE_DIR in master.conf (new location)
 echo ""
-echo "━━━ $ICON_GEAR Step 4: Update TARGET_DIR ━━━"
+echo "━━━ $ICON_GEAR Step 4: Update master.conf paths ━━━"
 NEW_MASTER="$DST/Configurations/master.conf"
 if [[ "$DRY_RUN" == false ]]; then
     if [[ -f "$NEW_MASTER" ]]; then
-        sed -i "s|^\(\s*TARGET_DIR\s*=\s*\)\"[^\"]*\"|\1\"$DST\"|" "$NEW_MASTER"
+        sed -i "s|^\(\s*TARGET_DIR\s*=\s*\)\"[^\"]*\"|\1\"$DST\"|"                              "$NEW_MASTER"
+        sed -i "s|^\(\s*DATA_DIR\s*=\s*\)\"[^\"]*\"|\1\"$DST/data\"|"                           "$NEW_MASTER"
+        sed -i "s|^\(\s*STATE_DIR\s*=\s*\)\"[^\"]*\"|\1\"$DST/State_Files\"|"                   "$NEW_MASTER"
         echo "  TARGET_DIR → $DST ✅"
+        echo "  DATA_DIR   → $DST/data ✅"
+        echo "  STATE_DIR  → $DST/State_Files ✅"
     else
         error "master.conf not found at $NEW_MASTER"
         exit 1
     fi
 else
-    warn "DRY RUN — would set TARGET_DIR=\"$DST\" in master.conf"
+    warn "DRY RUN — would update TARGET_DIR, DATA_DIR, STATE_DIR in master.conf"
 fi
 
 # ──────────────────────────────────────────────────────────────────────────────