diff --git a/models/behaviour/mail/oauth_v2.py b/models/behaviour/mail/oauth_v2.py
index c4b5240..3840bb5 100644
--- a/models/behaviour/mail/oauth_v2.py
+++ b/models/behaviour/mail/oauth_v2.py
@@ -222,25 +222,25 @@ class MailOAuthModel(BaseModel):
                 "_id": ObjectId(token_id),
                 "clientUserId": client_user_id
             }),
-            update = {
+            update = [{
                 "$set": {
                     "token": token,
                     "status": "active",
                     "lastRefreshTs": request_ts,
-                    # "firstRefreshTs": {
-                    #     "$cond": {
-                    #         "if": {
-                    #             "$or": [
-                    #                 {"$eq": ["$firstRefreshTs", None]},
-                    #                 {"$eq": [{"$type": "$firstRefreshTs"}, "missing"]}
-                    #             ]
-                    #         },
-                    #         "then": request_ts,
-                    #         "else": "$firstRefreshTs"
-                    #     }
-                    # }
+                    "firstRefreshTs": {
+                        "$cond": {
+                            "if": {
+                                "$or": [
+                                    {"$eq": ["$firstRefreshTs", None]},
+                                    {"$eq": [{"$type": "$firstRefreshTs"}, "missing"]}
+                                ]
+                            },
+                            "then": request_ts,
+                            "else": "$firstRefreshTs"
+                        }
+                    }
                 }
-            },
+            }],
             projection = {"token": False},
             return_updated = True,
             upsert = False