From 59ba32f0285aa6d6b367e6283cae9268e8947fed Mon Sep 17 00:00:00 2001
From: jeff
Date: Sat, 7 Mar 2015 16:00:44 -0500
Subject: match/unmatch button works

---
 src/cuchaz/enigma/convert/ClassMatch.java | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

(limited to 'src/cuchaz/enigma/convert/ClassMatch.java')

diff --git a/src/cuchaz/enigma/convert/ClassMatch.java b/src/cuchaz/enigma/convert/ClassMatch.java
index 9cecf70..eaaaa41 100644
--- a/src/cuchaz/enigma/convert/ClassMatch.java
+++ b/src/cuchaz/enigma/convert/ClassMatch.java
@@ -20,10 +20,16 @@ public class ClassMatch {
 	}
 	
 	public ClassMatch(ClassEntry sourceClass, ClassEntry destClass) {
-		this.sourceClasses = Sets.newHashSet(sourceClass);
-		this.destClasses = Sets.newHashSet(destClass);
+		sourceClasses = Sets.newHashSet();
+		if (sourceClass != null) {
+			sourceClasses.add(sourceClass);
+		}
+		destClasses = Sets.newHashSet();
+		if (destClass != null) {
+			destClasses.add(destClass);
+		}
 	}
-
+	
 	public boolean isMatched() {
 		return sourceClasses.size() > 0 && destClasses.size() > 0;
 	}
-- 
cgit v1.2.3