tweaks and improvements to field matching gui

1 files changed, 6 insertions, 4 deletions

diff --git a/src/cuchaz/enigma/analysis/SourceIndexBehaviorVisitor.java b/src/cuchaz/enigma/analysis/SourceIndexBehaviorVisitor.java
index a9a055b..eb120b6 100644
--- a/src/cuchaz/enigma/analysis/SourceIndexBehaviorVisitor.java
+++ b/src/cuchaz/enigma/analysis/SourceIndexBehaviorVisitor.java
@@ -111,10 +111,12 @@ public class SourceIndexBehaviorVisitor extends SourceIndexVisitor {
         @Override
         public Void visitParameterDeclaration(ParameterDeclaration node, SourceIndex index) {
                 ParameterDefinition def = node.getUserData(Keys.PARAMETER_DEFINITION);
-                MethodDefinition methodDef = (MethodDefinition)def.getMethod();
-                BehaviorEntry behaviorEntry = EntryFactory.getBehaviorEntry(methodDef);
-                ArgumentEntry argumentEntry = new ArgumentEntry(behaviorEntry, def.getPosition(), node.getName());
-                index.addDeclaration(node.getNameToken(), argumentEntry);
+                if (def.getMethod() instanceof MethodDefinition) {
+                        MethodDefinition methodDef = (MethodDefinition)def.getMethod();
+                        BehaviorEntry behaviorEntry = EntryFactory.getBehaviorEntry(methodDef);
+                        ArgumentEntry argumentEntry = new ArgumentEntry(behaviorEntry, def.getPosition(), node.getName());
+                        index.addDeclaration(node.getNameToken(), argumentEntry);
+                }
                 
                 return recurse(node, index);
         }