diff --git a/AlgebraicDataflowArchitectureModel/src/models/dataFlowModel/ModelExtension.java b/AlgebraicDataflowArchitectureModel/src/models/dataFlowModel/ModelExtension.java
index 8438c7b..06468f0 100644
--- a/AlgebraicDataflowArchitectureModel/src/models/dataFlowModel/ModelExtension.java
+++ b/AlgebraicDataflowArchitectureModel/src/models/dataFlowModel/ModelExtension.java
@@ -159,5 +159,7 @@
 		if (merge != null) merge.setMemento(mergeMem);		
 		Symbol extractFaceDown = model.getSymbol("extractFaceDown");
 		if (extractFaceDown != null) extractFaceDown.setMemento(extractFaceDownMem);
+		Symbol sortByKey = model.getSymbol("sortByKey");
+		if (sortByKey != null) sortByKey.setMemento(sortByKeyMem);
 	}
 }