1 files changed, 2 insertions, 1 deletions

diff --git a/src/test/java/cuchaz/enigma/TestDeobfuscator.java b/src/test/java/cuchaz/enigma/TestDeobfuscator.java
index 5f11721..e070b66 100644
--- a/src/test/java/cuchaz/enigma/TestDeobfuscator.java
+++ b/src/test/java/cuchaz/enigma/TestDeobfuscator.java
@@ -51,6 +51,7 @@ public class TestDeobfuscator {
         public void decompileClass()
                 throws Exception {
                 Deobfuscator deobfuscator = getDeobfuscator();
-                deobfuscator.getSource(deobfuscator.getSourceTree("a"));
+                SourceProvider sourceProvider = deobfuscator.getObfSourceProvider();
+                sourceProvider.writeSourceToString(sourceProvider.getSources("a"));
         }
 }