1 files changed, 11 insertions, 8 deletions

diff --git a/src/main/java/cuchaz/enigma/gui/GuiController.java b/src/main/java/cuchaz/enigma/gui/GuiController.java
index 5610233..1683333 100644
--- a/src/main/java/cuchaz/enigma/gui/GuiController.java
+++ b/src/main/java/cuchaz/enigma/gui/GuiController.java
@@ -297,14 +297,7 @@ public class GuiController {
          * @param reference
          */
         private void showReference(EntryReference<Entry<?>, Entry<?>> reference) {
-                EntryRemapper mapper = this.deobfuscator.getMapper();
-
-                SourceIndex index = this.currentSource.getIndex();
-                Collection<Token> tokens = mapper.getObfResolver().resolveReference(reference, ResolutionStrategy.RESOLVE_CLOSEST)
-                                .stream()
-                                .flatMap(r -> index.getReferenceTokens(r).stream())
-                                .collect(Collectors.toList());
-
+                Collection<Token> tokens = getTokensForReference(reference);
                 if (tokens.isEmpty()) {
                         // DEBUG
                         System.err.println(String.format("WARNING: no tokens found for %s in %s", reference, this.currentSource.getEntry()));
@@ -313,6 +306,16 @@ public class GuiController {
                 }
         }
 
+        public Collection<Token> getTokensForReference(EntryReference<Entry<?>, Entry<?>> reference) {
+                EntryRemapper mapper = this.deobfuscator.getMapper();
+
+                SourceIndex index = this.currentSource.getIndex();
+                return mapper.getObfResolver().resolveReference(reference, ResolutionStrategy.RESOLVE_CLOSEST)
+                                .stream()
+                                .flatMap(r -> index.getReferenceTokens(r).stream())
+                                .collect(Collectors.toList());
+        }
+
         public void openPreviousReference() {
                 if (hasPreviousReference()) {
                         setReference(gui.referenceHistory.goBack());