diff --git a/src/utils.jl b/src/utils.jl
index 752045e..0f5b4ff 100644
--- a/src/utils.jl
+++ b/src/utils.jl
@@ -232,7 +232,7 @@ end
 function checkTotalStepInPlan(a::agent)
   headers = []
 
-  for (k, v) in agent.memory[:shortterm]
+  for (k, v) in a.memory[:shortterm]
     push!(headers, k)
   end