diff --git a/src/store/reducer.ts b/src/store/reducer.ts
index d9b9fbb4..db291f42 100644
--- a/src/store/reducer.ts
+++ b/src/store/reducer.ts
@@ -137,7 +137,7 @@ export default function reactFlowReducer(state = initialState, action: ReactFlow
       const { id, diff, isDragging } = action.payload;
 
       const nextNodes = state.nodes.map((node) => {
-        if (id === node.id || state.selectedElements?.find((sNode) => sNode.id === node.id)) {
+        if (id === node.id || (id == null && state.selectedElements?.find((sNode) => sNode.id === node.id))) {
           const updatedNode = {
             ...node,
             __rf: {