diff --git a/arch/src/stream_context.rs b/arch/src/stream_context.rs
index c73728d8..4ef7a3c9 100644
--- a/arch/src/stream_context.rs
+++ b/arch/src/stream_context.rs
@@ -746,7 +746,7 @@ impl StreamContext {
             if arch_assistant {
                 for message in messages.iter() {
                     if let Some(model) = message.model.as_ref() {
-                        if model.contains("gpt") {
+                        if !model.contains("Arch") {
                             break;
                         }
                     }