diff --git a/frontend/src/components/visual-editor/v2/Diagrams.tsx b/frontend/src/components/visual-editor/v2/Diagrams.tsx
index 34ae954..3783755 100644
--- a/frontend/src/components/visual-editor/v2/Diagrams.tsx
+++ b/frontend/src/components/visual-editor/v2/Diagrams.tsx
@@ -453,28 +453,12 @@ const Diagrams = () => {
 
     if (ids) {
       for (const blockId of ids) {
-        const block = getBlockFromCache(blockId);
-
-        await updateBlock(
-          {
-            id: blockId,
-            params: {
-              category: newCategoryId,
-            },
+        await updateBlock({
+          id: blockId,
+          params: {
+            category: newCategoryId,
           },
-          {
-            onSuccess() {
-              updateCachedBlock({
-                id: blockId,
-                payload: {
-                  ...block,
-                  category: newCategoryId,
-                },
-                strategy: "overwrite",
-              });
-            },
-          },
-        );
+        });
       }
 
       queryClient.removeQueries({