diff --git a/PacletInfo.wl b/PacletInfo.wl
index 3b6afaa4..c854b31c 100644
--- a/PacletInfo.wl
+++ b/PacletInfo.wl
@@ -1,7 +1,7 @@
 PacletObject[ <|
     "Name"           -> "Wolfram/Chatbook",
     "PublisherID"    -> "Wolfram",
-    "Version"        -> "2.1.0",
+    "Version"        -> "2.1.1",
     "WolframVersion" -> "14.1+",
     "Description"    -> "Wolfram Notebooks + LLMs",
     "License"        -> "MIT",
diff --git a/Source/Chatbook/Actions.wl b/Source/Chatbook/Actions.wl
index d05b332e..5e2c226a 100644
--- a/Source/Chatbook/Actions.wl
+++ b/Source/Chatbook/Actions.wl
@@ -739,8 +739,8 @@ StopChat[ cell0_CellObject ] := Enclose[
         settings = ConfirmMatch[ currentChatSettings @ cell, _Association|_Missing, "ChatNotebookSettings" ];
         If[ MissingQ @ settings, finish[ ] ];
         removeTask @ Lookup[ settings, "Task", None ];
-        container = ConfirmMatch[ Lookup[ settings, "Container", None ], _Association|None, "Container" ];
-        If[ container === None, finish[ ] ];
+        container = ConfirmMatch[ Lookup[ settings, "Container", None ], _Association|_Symbol, "Container" ];
+        If[ MatchQ[ container, _Symbol ], finish[ ] ];
         content = ConfirmMatch[ Lookup[ container, "FullContent" ], _String|$$progressIndicator, "Content" ];
         FinishDynamic[ ];
         Block[ { createFETask = # & }, writeReformattedCell[ settings, content, cell ] ]