diff --git a/pkg/channels/wecom/aibot_test.go b/pkg/channels/wecom/aibot_test.go
index 01f3d3a41..957b51c38 100644
--- a/pkg/channels/wecom/aibot_test.go
+++ b/pkg/channels/wecom/aibot_test.go
@@ -146,10 +146,14 @@ func TestWeComAIBotChannelGetStreamResponseProcessingMessage(t *testing.T) {
 		}
 
 		messageBus := bus.NewMessageBus()
-		ch, err := NewWeComAIBotChannel(cfg, messageBus)
+		channel, err := NewWeComAIBotChannel(cfg, messageBus)
 		if err != nil {
 			t.Fatalf("Failed to create channel: %v", err)
 		}
+		ch, ok := channel.(*WeComAIBotChannel)
+		if !ok {
+			t.Fatal("Expected webhook mode channel")
+		}
 
 		task := &streamTask{
 			StreamID: "stream-default",
@@ -189,10 +193,14 @@ func TestWeComAIBotChannelGetStreamResponseProcessingMessage(t *testing.T) {
 		}
 
 		messageBus := bus.NewMessageBus()
-		ch, err := NewWeComAIBotChannel(cfg, messageBus)
+		channel, err := NewWeComAIBotChannel(cfg, messageBus)
 		if err != nil {
 			t.Fatalf("Failed to create channel: %v", err)
 		}
+		ch, ok := channel.(*WeComAIBotChannel)
+		if !ok {
+			t.Fatal("Expected webhook mode channel")
+		}
 
 		task := &streamTask{
 			StreamID: "stream-custom",