Little tweak with the agent system