diff --git a/packages/stream_chat/test/src/core/api/message_api_test.dart b/packages/stream_chat/test/src/core/api/message_api_test.dart
index ed7fefd6..8ae82e94 100644
--- a/packages/stream_chat/test/src/core/api/message_api_test.dart
+++ b/packages/stream_chat/test/src/core/api/message_api_test.dart
@@ -394,7 +394,12 @@ void main() {
           path,
           data: {'language': language},
         )).thenAnswer((_) async => successResponse(path, data: {
-          'message': translatedMessage.toJson(),
+          'message': {
+            ...translatedMessage.toJson(),
+            'i18n': {
+              language: translatedMessageText,
+            },
+          },
         }));
 
     final res = await messageApi.translateMessage(messageId, language);