diff --git a/com.unity.ml-agents/Runtime/Agent.cs b/com.unity.ml-agents/Runtime/Agent.cs
index 5a37347b31..5eb73d8330 100644
--- a/com.unity.ml-agents/Runtime/Agent.cs
+++ b/com.unity.ml-agents/Runtime/Agent.cs
@@ -586,7 +586,6 @@ void NotifyAgentDone(DoneReason doneReason)
             // Request the last decision with no callbacks
             // We request a decision so Python knows the Agent is done immediately
             m_Brain?.RequestDecision(m_Info, sensors);
-            ResetSensors();
 
             // We also have to write any to any DemonstationStores so that they get the "done" flag.
             foreach (var demoWriter in DemonstrationWriters)
@@ -594,6 +593,8 @@ void NotifyAgentDone(DoneReason doneReason)
                 demoWriter.Record(m_Info, sensors);
             }
 
+            ResetSensors();
+
             if (doneReason != DoneReason.Disabled)
             {
                 // We don't want to update the reward stats when the Agent is disabled, because this will make