diff --git a/coderd/x/chatd/subagent_internal_test.go b/coderd/x/chatd/subagent_internal_test.go
index 1d2aeb86fc..4ab9ab431e 100644
--- a/coderd/x/chatd/subagent_internal_test.go
+++ b/coderd/x/chatd/subagent_internal_test.go
@@ -4237,7 +4237,8 @@ func TestAwaitSubagentCompletion(t *testing.T) {
 		_, _, err = server.awaitSubagentCompletion(
 			shortCtx, parent.ID, child.ID, 5*time.Second,
 		)
-		require.ErrorIs(t, err, context.DeadlineExceeded)
+		require.ErrorIs(t, shortCtx.Err(), context.DeadlineExceeded)
+		require.True(t, database.IsQueryCanceledError(err))
 	})
 
 	t.Run("ZeroTimeoutUsesDefault", func(t *testing.T) {