1 files changed, 12 insertions, 6 deletions

diff --git a/src/cuchaz/enigma/analysis/SourceIndexVisitor.java b/src/cuchaz/enigma/analysis/SourceIndexVisitor.java
index 6c14ee9..f31eb1a 100644
--- a/src/cuchaz/enigma/analysis/SourceIndexVisitor.java
+++ b/src/cuchaz/enigma/analysis/SourceIndexVisitor.java
@@ -201,6 +201,18 @@ public class SourceIndexVisitor implements IAstVisitor<SourceIndex, Void>
                 return recurse( node, index );
         }
         
+        @Override
+        public Void visitEnumValueDeclaration( EnumValueDeclaration node, SourceIndex index )
+        {
+                // treat enum declarations as field declarations
+                FieldDefinition def = node.getUserData( Keys.FIELD_DEFINITION );
+                ClassEntry classEntry = new ClassEntry( def.getDeclaringType().getInternalName() );
+                FieldEntry fieldEntry = new FieldEntry( classEntry, def.getName() );
+                index.addDeclaration( node.getNameToken(), fieldEntry );
+                
+                return recurse( node, index );
+        }
+        
         private Void recurse( AstNode node, SourceIndex index )
         {
                 for( final AstNode child : node.getChildren() )
@@ -561,12 +573,6 @@ public class SourceIndexVisitor implements IAstVisitor<SourceIndex, Void>
         }
         
         @Override
-        public Void visitEnumValueDeclaration( EnumValueDeclaration node, SourceIndex index )
-        {
-                return recurse( node, index );
-        }
-        
-        @Override
         public Void visitAssertStatement( AssertStatement node, SourceIndex index )
         {
                 return recurse( node, index );