1 files changed, 4 insertions, 0 deletions
diff --git a/src/gui/undocommands.cpp b/src/gui/undocommands.cpp
index e2502f1..9c52bf3 100644
--- a/src/gui/undocommands.cpp
+++ b/src/gui/undocommands.cpp
@@ -421,12 +421,16 @@ ReplaceGraphCommand::ReplaceGraphCommand(TikzScene *scene, Graph *oldGraph, Grap
 
 void ReplaceGraphCommand::undo()
 {
+    foreach (Node *n, _oldGraph->nodes()) n->attachStyle();
+    foreach (Edge *e, _oldGraph->edges()) e->updateControls();
     _scene->tikzDocument()->setGraph(_oldGraph);
     _scene->graphReplaced();
 }
 
 void ReplaceGraphCommand::redo()
 {
+    foreach (Node *n, _newGraph->nodes()) n->attachStyle();
+    foreach (Edge *e, _newGraph->edges()) e->updateControls();
     _scene->tikzDocument()->setGraph(_newGraph);
     _scene->graphReplaced();
 }