diff --git a/coderd/exp_chats_test.go b/coderd/exp_chats_test.go
index a41d365524..3686b9795a 100644
--- a/coderd/exp_chats_test.go
+++ b/coderd/exp_chats_test.go
@@ -7290,7 +7290,7 @@ func TestPatchChatMessage(t *testing.T) {
 		overrideModel := createAdditionalChatModelConfig(
 			t,
 			client,
-			"openai",
+			defaultModel.Provider,
 			"gpt-4o-mini-edit-override",
 		)
 
diff --git a/site/src/pages/AgentsPage/components/ChatConversation/messageHelpers.test.ts b/site/src/pages/AgentsPage/components/ChatConversation/messageHelpers.test.ts
index b2127454c8..8add3c9b0a 100644
--- a/site/src/pages/AgentsPage/components/ChatConversation/messageHelpers.test.ts
+++ b/site/src/pages/AgentsPage/components/ChatConversation/messageHelpers.test.ts
@@ -19,6 +19,7 @@ const getDisplayState = (message: ChatMessage) =>
 		message,
 		parsed: parseMessageContent(message.content),
 		hideActions: false,
+		hasActiveStream: false,
 	});
 
 describe("deriveMessageDisplayState", () => {