1 files changed, 10 insertions, 2 deletions
diff --git a/plugins/infra/gmfdiag/org.eclipse.papyrus.infra.gmfdiag.common/src/org/eclipse/papyrus/infra/gmfdiag/common/helper/DiagramPrototype.java b/plugins/infra/gmfdiag/org.eclipse.papyrus.infra.gmfdiag.common/src/org/eclipse/papyrus/infra/gmfdiag/common/helper/DiagramPrototype.java
index e28a4eaf4ec..3c8632a9d1b 100755
--- a/plugins/infra/gmfdiag/org.eclipse.papyrus.infra.gmfdiag.common/src/org/eclipse/papyrus/infra/gmfdiag/common/helper/DiagramPrototype.java
+++ b/plugins/infra/gmfdiag/org.eclipse.papyrus.infra.gmfdiag.common/src/org/eclipse/papyrus/infra/gmfdiag/common/helper/DiagramPrototype.java
@@ -82,10 +82,14 @@ public class DiagramPrototype extends ViewPrototype {
 				DiagramUtils.setOwner(diagram, target);
 			}
 			@Override
-			public void redo() {
+			public void undo() {
 				DiagramUtils.setOwner(diagram, previous);
 			}
 			@Override
+			public void redo() {
+				DiagramUtils.setOwner(diagram, target);
+			}
+			@Override
 			protected boolean prepare() { return true; }
 		};
 	}
@@ -100,10 +104,14 @@ public class DiagramPrototype extends ViewPrototype {
 				diagram.setElement(target);
 			}
 			@Override
-			public void redo() {
+			public void undo() {
 				diagram.setElement(previous);
 			}
 			@Override
+			public void redo() {
+				diagram.setElement(target);
+			}
+			@Override
 			protected boolean prepare() { return true; }
 		};
 	}