diff --git a/all_code.txt b/all_code.txt
deleted file mode 100644
index 8b137891..00000000
--- a/all_code.txt
+++ /dev/null
@@ -1 +0,0 @@
-
diff --git a/src/controlflow/events/events.py b/src/controlflow/events/events.py
index 90b6910c..74eaa2bf 100644
--- a/src/controlflow/events/events.py
+++ b/src/controlflow/events/events.py
@@ -101,6 +101,7 @@ def _message(cls, v):
     def _finalize(self):
         self.delta["name"] = self.agent.name
         self.snapshot["name"] = self.agent.name
+        return self
 
     @property
     def delta_message(self) -> AIMessageChunk: