diff --git a/packages/stream_chat/test/src/api/channel_test.dart b/packages/stream_chat/test/src/api/channel_test.dart
index f92b05bb..6d15f4ef 100644
--- a/packages/stream_chat/test/src/api/channel_test.dart
+++ b/packages/stream_chat/test/src/api/channel_test.dart
@@ -430,7 +430,7 @@ void main() {
         await channelClient.watch();
 
         when(
-          () => mockDio.post<String>(
+          () => mockDio.put<String>(
             '/messages/${message.id}',
             data: anything,
           ),
@@ -445,7 +445,7 @@ void main() {
         await channelClient.pinMessage(message, 30);
 
         verify(() =>
-                mockDio.post<String>('/messages/${message.id}', data: anything))
+                mockDio.put<String>('/messages/${message.id}', data: anything))
             .called(1);
       });
 
@@ -475,7 +475,7 @@ void main() {
         await channelClient.watch();
 
         when(
-          () => mockDio.post<String>(
+          () => mockDio.put<String>(
             '/messages/${message.id}',
             data: anything,
           ),
@@ -490,7 +490,7 @@ void main() {
         await channelClient.pinMessage(message);
 
         verify(() =>
-                mockDio.post<String>('/messages/${message.id}', data: anything))
+                mockDio.put<String>('/messages/${message.id}', data: anything))
             .called(1);
       });
 
@@ -520,7 +520,7 @@ void main() {
         await channelClient.watch();
 
         when(
-          () => mockDio.post<String>(
+          () => mockDio.put<String>(
             '/messages/${message.id}',
             data: anything,
           ),
@@ -535,7 +535,7 @@ void main() {
         await channelClient.unpinMessage(message);
 
         verify(() =>
-                mockDio.post<String>('/messages/${message.id}', data: anything))
+                mockDio.put<String>('/messages/${message.id}', data: anything))
             .called(1);
       });
     });
diff --git a/packages/stream_chat/test/src/client_test.dart b/packages/stream_chat/test/src/client_test.dart
index 53792764..431b0c04 100644
--- a/packages/stream_chat/test/src/client_test.dart
+++ b/packages/stream_chat/test/src/client_test.dart
@@ -745,6 +745,39 @@ void main() {
             data: {'message': anything})).called(1);
       });
 
+      test('partiallyUpdateMessage', () async {
+        final mockDio = MockDio();
+
+        when(() => mockDio.options).thenReturn(BaseOptions());
+        when(() => mockDio.interceptors).thenReturn(Interceptors());
+
+        final client = StreamChatClient('api-key', httpClient: mockDio);
+        final message = Message(
+          id: 'test',
+          text: 'demo',
+        );
+
+        when(
+          () => mockDio.put<String>(
+            '/messages/${message.id}',
+            data: {'set': anything},
+          ),
+        ).thenAnswer(
+          (_) async => Response(
+            data: jsonEncode({'message': message}),
+            statusCode: 200,
+            requestOptions: FakeRequestOptions(),
+          ),
+        );
+
+        await client.partiallyUpdateMessage(message.id, {
+          'set': {'text': message.text}
+        });
+
+        verify(() => mockDio.put<String>('/messages/${message.id}',
+            data: {'set': anything})).called(1);
+      });
+
       test('deleteMessage', () async {
         final mockDio = MockDio();
 
@@ -1094,7 +1127,7 @@ void main() {
           final message = Message(text: 'Hello');
 
           when(
-            () => mockDio.post<String>(
+            () => mockDio.put<String>(
               '/messages/${message.id}',
               data: anything,
             ),
@@ -1108,7 +1141,7 @@ void main() {
 
           await client.pinMessage(message, timeout);
 
-          verify(() => mockDio.post<String>('/messages/${message.id}',
+          verify(() => mockDio.put<String>('/messages/${message.id}',
               data: {'message': anything})).called(1);
         });