2 files changed, 5 insertions, 6 deletions

diff --git a/enigma/src/main/java/cuchaz/enigma/source/procyon/transformers/DropVarModifiersAstTransform.java b/enigma/src/main/java/cuchaz/enigma/source/procyon/transformers/DropVarModifiersAstTransform.java
index defd251..14b5cc0 100644
--- a/enigma/src/main/java/cuchaz/enigma/source/procyon/transformers/DropVarModifiersAstTransform.java
+++ b/enigma/src/main/java/cuchaz/enigma/source/procyon/transformers/DropVarModifiersAstTransform.java
@@ -1,7 +1,6 @@
 package cuchaz.enigma.source.procyon.transformers;
 
-import javax.lang.model.element.Modifier;
-
+import com.strobel.assembler.metadata.Flags;
 import com.strobel.decompiler.languages.java.ast.AstNode;
 import com.strobel.decompiler.languages.java.ast.DepthFirstAstVisitor;
 import com.strobel.decompiler.languages.java.ast.EntityDeclaration;
@@ -25,7 +24,7 @@ public final class DropVarModifiersAstTransform implements IAstTransform {
                 @Override
                 public Void visitParameterDeclaration(ParameterDeclaration node, Void data) {
                         for (JavaModifierToken modifierToken : node.getChildrenByRole(EntityDeclaration.MODIFIER_ROLE)) {
-                                if (modifierToken.getModifier() == Modifier.FINAL) {
+                                if (modifierToken.getModifier() == Flags.Flag.FINAL) {
                                         modifierToken.remove();
                                 }
                         }
@@ -35,7 +34,7 @@ public final class DropVarModifiersAstTransform implements IAstTransform {
 
                 @Override
                 public Void visitVariableDeclaration(VariableDeclarationStatement node, Void data) {
-                        node.removeModifier(Modifier.FINAL);
+                        node.removeModifier(Flags.Flag.FINAL);
                         return null;
                 }
         }
diff --git a/enigma/src/main/java/cuchaz/enigma/source/procyon/transformers/ObfuscatedEnumSwitchRewriterTransform.java b/enigma/src/main/java/cuchaz/enigma/source/procyon/transformers/ObfuscatedEnumSwitchRewriterTransform.java
index 204351e..4f34b33 100644
--- a/enigma/src/main/java/cuchaz/enigma/source/procyon/transformers/ObfuscatedEnumSwitchRewriterTransform.java
+++ b/enigma/src/main/java/cuchaz/enigma/source/procyon/transformers/ObfuscatedEnumSwitchRewriterTransform.java
@@ -89,7 +89,7 @@ public class ObfuscatedEnumSwitchRewriterTransform implements IAstTransform {
                 }
 
                 @Override
-                public Void visitTypeDeclaration(final TypeDeclaration typeDeclaration, final Void p) {
+                public Void visitTypeDeclarationOverride(final TypeDeclaration typeDeclaration, final Void p) {
                         final boolean oldIsSwitchMapWrapper = _isSwitchMapWrapper;
                         final TypeDefinition typeDefinition = typeDeclaration.getUserData(Keys.TYPE_DEFINITION);
                         final boolean isSwitchMapWrapper = isSwitchMapWrapper(typeDefinition);
@@ -109,7 +109,7 @@ public class ObfuscatedEnumSwitchRewriterTransform implements IAstTransform {
                         _isSwitchMapWrapper = isSwitchMapWrapper;
 
                         try {
-                                super.visitTypeDeclaration(typeDeclaration, p);
+                                super.visitTypeDeclarationOverride(typeDeclaration, p);
                         } finally {
                                 _isSwitchMapWrapper = oldIsSwitchMapWrapper;
                         }