1 files changed, 1 insertions, 1 deletions

diff --git a/src/main/java/cuchaz/enigma/gui/CodeReader.java b/src/main/java/cuchaz/enigma/gui/CodeReader.java
index f76dc89..97deaf3 100644
--- a/src/main/java/cuchaz/enigma/gui/CodeReader.java
+++ b/src/main/java/cuchaz/enigma/gui/CodeReader.java
@@ -162,7 +162,7 @@ public class CodeReader extends JEditorPane {
                         // couldn't find the class declaration token, might be an anonymous class
                         // look for any declaration in that class instead
                         for (Entry entry : sourceIndex.declarations()) {
-                                if (entry.getClassEntry().equals(classEntry)) {
+                                if (entry.getOwnerClassEntry().equals(classEntry)) {
                                         token = sourceIndex.getDeclarationToken(entry);
                                         break;
                                 }