1 files changed, 4 insertions, 4 deletions
diff --git a/st/MethodInstance.java b/st/MethodInstance.java
index c2f978b..9f1e6df 100644
--- a/st/MethodInstance.java
+++ b/st/MethodInstance.java
@@ -5,10 +5,10 @@ import java.util.ArrayList;
 public class MethodInstance extends AbstractInstance {
     private ArrayList<TypeInstance> args;       // the list of arguments
     private ArrayList<TypeInstance> lvars;      // the list of local variables
-    protected ClassInstance par_cls;                // the surrounding class
-    private TypeEnum rtrn;                      // the returned type
+    protected ClassInstance par_cls;            // the surrounding class
+    private ClassInstance rtrn;                      // the returned type
 
-    public MethodInstance(String name, TypeEnum rtrn, ClassInstance par_cls) {
+    public MethodInstance(String name, ClassInstance rtrn, ClassInstance par_cls) {
         super(name, TypeEnum.method);
         this.lvars = new ArrayList<>();
         this.args = new ArrayList<>();
@@ -31,7 +31,7 @@ public class MethodInstance extends AbstractInstance {
         return this.lvars;
     }
 
-    public TypeEnum getReturn() {
+    public ClassInstance getReturn() {
         return this.rtrn;
     }