1 files changed, 4 insertions, 4 deletions

diff --git a/src/test/java/cuchaz/enigma/TokenChecker.java b/src/test/java/cuchaz/enigma/TokenChecker.java
index c4670a2..9e0c696 100644
--- a/src/test/java/cuchaz/enigma/TokenChecker.java
+++ b/src/test/java/cuchaz/enigma/TokenChecker.java
@@ -25,16 +25,16 @@ import java.util.jar.JarFile;
 
 public class TokenChecker {
 
-        private Deobfuscator deobfuscator;
+        private Enigma enigma;
 
         protected TokenChecker(JarFile jarFile)
                 throws IOException {
-                deobfuscator = new Deobfuscator(jarFile);
+                enigma = new Enigma(jarFile);
         }
 
         protected String getDeclarationToken(Entry<?> entry) {
                 // decompile the class
-                SourceProvider sourceProvider = deobfuscator.getObfSourceProvider();
+                SourceProvider sourceProvider = enigma.getObfSourceProvider();
                 CompilationUnit tree = sourceProvider.getSources(entry.getContainingClass().getFullName());
                 // DEBUG
                 // tree.acceptVisitor( new TreeDumpVisitor( new File( "tree." + entry.getClassName().replace( '/', '.' ) + ".txt" ) ), null );
@@ -52,7 +52,7 @@ public class TokenChecker {
         @SuppressWarnings("unchecked")
         protected Collection<String> getReferenceTokens(EntryReference<? extends Entry<?>, ? extends Entry<?>> reference) {
                 // decompile the class
-                SourceProvider sourceProvider = deobfuscator.getObfSourceProvider();
+                SourceProvider sourceProvider = enigma.getObfSourceProvider();
                 CompilationUnit tree = sourceProvider.getSources(reference.context.getContainingClass().getFullName());
                 String source = sourceProvider.writeSourceToString(tree);
                 SourceIndex index = SourceIndex.buildIndex(source, tree, true);