ui improvements

1 files changed, 6 insertions, 2 deletions

diff --git a/src/cuchaz/enigma/convert/ClassForest.java b/src/cuchaz/enigma/convert/ClassForest.java
index 3673139..a5ea056 100644
--- a/src/cuchaz/enigma/convert/ClassForest.java
+++ b/src/cuchaz/enigma/convert/ClassForest.java
@@ -24,8 +24,12 @@ public class ClassForest {
                 }
         }
 
-        private void add(ClassEntry entry) {
-                m_forest.put(m_identifier.identify(entry), entry);
+        public void add(ClassEntry entry) {
+                try {
+                        m_forest.put(m_identifier.identify(entry), entry);
+                } catch (ClassNotFoundException ex) {
+                        throw new Error("Unable to find class " + entry.getName());
+                }
         }
 
         public Collection<ClassIdentity> identities() {