diff --git a/src/main/java/edu/lu/uni/serval/gumtree/regroup/SimplifyTree.java b/src/main/java/edu/lu/uni/serval/gumtree/regroup/SimplifyTree.java
index 33fcfc1..b43756b 100644
--- a/src/main/java/edu/lu/uni/serval/gumtree/regroup/SimplifyTree.java
+++ b/src/main/java/edu/lu/uni/serval/gumtree/regroup/SimplifyTree.java
@@ -147,7 +147,11 @@ public class SimplifyTree {
 				}
 			} else {
 				simpleTree.setNodeType(astNode);
-				simpleTree.setLabel(label.replaceAll(" ", ""));
+				if (astNode.endsWith("Type")) {
+					simpleTree.setLabel(canonicalizeTypeStr(label).replaceAll(" ", ""));
+				} else {
+					simpleTree.setLabel(label.replaceAll(" ", ""));
+				}
 			}
 		}