From e0b4b7c47f07a193e96a058ffa23eff47cbdceab Mon Sep 17 00:00:00 2001
From: Kui LIU <brucekuiliu@gmail.com>
Date: Mon, 31 Jul 2017 18:12:26 +0200
Subject: [PATCH] Simplify SimpleType.

---
 .../edu/lu/uni/serval/gumtree/regroup/SimplifyTree.java     | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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(" ", ""));
+				}
 			}
 		}