diff --git a/airflow/models/taskinstance.py b/airflow/models/taskinstance.py
index 30fb9b5eb7531..4a378ddb18f8d 100644
--- a/airflow/models/taskinstance.py
+++ b/airflow/models/taskinstance.py
@@ -1522,7 +1522,7 @@ def _run_raw_task(
 
         if not test_mode:
             session.add(Log(self.state, self))
-            session.merge(self)
+            session.merge(self).task = self.task
             if self.state == TaskInstanceState.SUCCESS:
                 self._register_dataset_changes(session=session)
             session.commit()