Updated code

routes/profileUpdate.js

@@ -197,7 +197,7 @@ router.put("/:userId/addresses/:addressId", asyncHandler(async (req, res) => {
         const updatedUser = await User.findOneAndUpdate(
             {
                 unique_id: req.params.userId,
-                "addresses._id": req.params.addressId
+                "addresses._id": req.params.addressId // Compare as string
             },
             {
                 $set: {
@@ -208,14 +208,23 @@ router.put("/:userId/addresses/:addressId", asyncHandler(async (req, res) => {
         );
 
         if (!updatedUser) {
-            return res.status(404).json(failedResponse("Address not found", 404));
+            return res.status(404).json(failedResponse("User or address not found", 404));
         }
-        console.log(JSON.stringify(updatedUser))
-        const updatedAddress = updatedUser.addresses.find(a => a._id.equals(req.params.addressId));
-        console.log("FINALL LIST: "+JSON.stringify(updatedUser))
+
+        // Find the updated address by comparing _id as a string
+        const updatedAddress = updatedUser.addresses.find(a => a._id === req.params.addressId);
+
+        if (!updatedAddress) {
+            return res.status(404).json(failedResponse("Address not found in the user's addresses", 404));
+        }
+
+        console.log("Updated User:", updatedUser);
+        console.log("Updated Address:", updatedAddress);
+
         res.status(200).json(successResponse("Address updated successfully", updatedAddress));
     } catch (error) {
-        return res.status(500).json(failedResponse(error, 500));
+        console.error("Error updating address:", error);
+        return res.status(500).json(failedResponse(error.message || "Internal server error", 500));
     }
 }));