diff --git a/src/index.ts b/src/index.ts
index 2eb9c26..c7f4051 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -117,6 +117,10 @@ export default function skillReinject(pi: ExtensionAPI): void {
 		restoreSessionState(ctx);
 	});
 
+	pi.on("session_shutdown", async () => {
+		persistState();
+	});
+
 	pi.on("session_before_compact", async () => {
 		markAutoCompactionBeforeCompact(compactionRuntime);
 	});