added types to fields

1 files changed, 3 insertions, 2 deletions

diff --git a/src/cuchaz/enigma/bytecode/ClassTranslator.java b/src/cuchaz/enigma/bytecode/ClassTranslator.java
index afd3a77..4dba0d8 100644
--- a/src/cuchaz/enigma/bytecode/ClassTranslator.java
+++ b/src/cuchaz/enigma/bytecode/ClassTranslator.java
@@ -55,7 +55,8 @@ public class ClassTranslator {
                                         // translate the name
                                         FieldEntry entry = new FieldEntry(
                                                 new ClassEntry(Descriptor.toJvmName(constants.getFieldrefClassName(i))),
-                                                constants.getFieldrefName(i)
+                                                constants.getFieldrefName(i),
+                                                new Type(constants.getFieldrefType(i))
                                         );
                                         FieldEntry translatedEntry = m_translator.translateEntry(entry);
                                         
@@ -94,7 +95,7 @@ public class ClassTranslator {
                 for (CtField field : c.getDeclaredFields()) {
                         
                         // translate the name
-                        FieldEntry entry = new FieldEntry(classEntry, field.getName());
+                        FieldEntry entry = JavassistUtil.getFieldEntry(field);
                         String translatedName = m_translator.translate(entry);
                         if (translatedName != null) {
                                 field.setName(translatedName);