From 9b105557d8170b458a3b03570b5c96f806f19ce1 Mon Sep 17 00:00:00 2001
From: suspiciousRaccoon <127566947+suspiciousRaccoon@users.noreply.github.com>
Date: Mon, 12 May 2025 23:25:35 -0300
Subject: [PATCH] fix: deprecation warning wihen using sqlmodel_update

---
 sqlmodel/main.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/sqlmodel/main.py b/sqlmodel/main.py
index 38c85915aa..c7ad166184 100644
--- a/sqlmodel/main.py
+++ b/sqlmodel/main.py
@@ -992,17 +992,17 @@ def sqlmodel_update(
         use_update = (update or {}).copy()
         if isinstance(obj, dict):
             for key, value in {**obj, **use_update}.items():
-                if key in get_model_fields(self):
+                if key in get_model_fields(self.__class__):
                     setattr(self, key, value)
         elif isinstance(obj, BaseModel):
-            for key in get_model_fields(obj):
+            for key in get_model_fields(obj.__class__):
                 if key in use_update:
                     value = use_update.pop(key)
                 else:
                     value = getattr(obj, key)
                 setattr(self, key, value)
             for remaining_key in use_update:
-                if remaining_key in get_model_fields(self):
+                if remaining_key in get_model_fields(self.__class__):
                     value = use_update.pop(remaining_key)
                     setattr(self, remaining_key, value)
         else: