diff --git a/NodeGraphQt/base/commands.py b/NodeGraphQt/base/commands.py
index a542fde9..966db5cc 100644
--- a/NodeGraphQt/base/commands.py
+++ b/NodeGraphQt/base/commands.py
@@ -118,7 +118,7 @@ def undo(self):
         self.node.view.delete()
 
     def redo(self):
-        self.graph.model.nodes[self.node.id] = self.node
+        self.model.nodes[self.node.id] = self.node
         self.viewer.add_node(self.node.view, self.pos)