diff --git a/client/coral-framework/graphql/mutations.js b/client/coral-framework/graphql/mutations.js
index dcba97891..3b9ef01f4 100644
--- a/client/coral-framework/graphql/mutations.js
+++ b/client/coral-framework/graphql/mutations.js
@@ -200,8 +200,8 @@ export const withAddTag = withMutation(
             asset_id: assetId
           },
           optimisticResponse: {
-            deleteAction: {
-              __typename: 'DeleteActionResponse',
+            addTag: {
+              __typename: 'ModifyTagResponse',
               errors: null,
             }
           },
@@ -242,6 +242,12 @@ export const withRemoveTag = withMutation(
             name,
             asset_id: assetId
           },
+          optimisticResponse: {
+            removeTag: {
+              __typename: 'ModifyTagResponse',
+              errors: null,
+            }
+          },
           update: (proxy) => {
             const fragmentId = `Comment_${id}`;