diff --git a/packages/stream_chat_flutter/test/src/channel_header_test.dart b/packages/stream_chat_flutter/test/src/channel_header_test.dart
index 6590ee66..831809a7 100644
--- a/packages/stream_chat_flutter/test/src/channel_header_test.dart
+++ b/packages/stream_chat_flutter/test/src/channel_header_test.dart
@@ -65,7 +65,7 @@ void main() {
       ));
 
       expect(find.text('test'), findsOneWidget);
-      expect(find.byType(ChannelAvatar), findsOneWidget);
+      expect(find.byType(StreamChannelAvatar), findsOneWidget);
       expect(find.byType(StreamBackButton), findsOneWidget);
       expect(find.byType(StreamChannelInfo), findsOneWidget);
     },
@@ -428,8 +428,8 @@ void main() {
       ));
 
       await tester.tap(find.byType(StreamBackButton));
-      await tester.tap(find.byType(ChannelAvatar));
-      await tester.tap(find.byType(ChannelName));
+      await tester.tap(find.byType(StreamChannelAvatar));
+      await tester.tap(find.byType(StreamChannelName));
 
       expect(backPressed, true);
       expect(imageTapped, true);
diff --git a/packages/stream_chat_flutter/test/src/channel_preview_test.dart b/packages/stream_chat_flutter/test/src/channel_preview_test.dart
index 47554605..285a5373 100644
--- a/packages/stream_chat_flutter/test/src/channel_preview_test.dart
+++ b/packages/stream_chat_flutter/test/src/channel_preview_test.dart
@@ -83,7 +83,7 @@ void main() {
       expect(find.text('test name'), findsOneWidget);
       expect(find.text('1'), findsOneWidget);
       expect(find.text('hello'), findsOneWidget);
-      expect(find.byType(ChannelAvatar), findsOneWidget);
+      expect(find.byType(StreamChannelAvatar), findsOneWidget);
     },
   );
 }