ACP2E-2791: Not able to Save Customer attribute information in Admin Edit customer section;

Author: pradeep1819

app/code/Magento/Customer/Model/SetCustomerStore.php

@@ -39,16 +39,17 @@ public function __construct(private StoreManagerInterface $storeManager)
      */
     public function setStore(array|null $requestData = null): void
     {
-        $storeId = $requestData[CustomerInterface::STORE_ID] ?? null;
+        $websiteId = $requestData[CustomerInterface::WEBSITE_ID] ?? null;
+        try {
+            $website = $this->storeManager->getWebsite($websiteId);
+            $storeId = $website ? current($website->getStoreIds()) : null;
+        } catch (LocalizedException $e) {
+            $storeId = null;
+        }
         if (!$storeId) {
-            $websiteId = $requestData[CustomerInterface::WEBSITE_ID] ?? null;
-            try {
-                $website = $this->storeManager->getWebsite($websiteId);
-                $storeId = $website ? current($website->getStoreIds()) : null;
-            } catch (LocalizedException $e) {
-                $storeId = null;
-            }
+            $storeId = $requestData[CustomerInterface::STORE_ID] ?? null;
         }
+
         $this->storeManager->setCurrentStore($storeId);
     }
 }