From e5709fbc956d3395e7b401fd5198524e02542e81 Mon Sep 17 00:00:00 2001
From: bd-912 <bdunahu@colostate.edu>
Date: Sun, 12 May 2024 13:30:35 -0600
Subject: Make (classinstance).send typecheck in HeatVisitor

---
 heat/HeatVisitor.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'heat/HeatVisitor.java')

diff --git a/heat/HeatVisitor.java b/heat/HeatVisitor.java
index 41fd3fe..e02f0d2 100644
--- a/heat/HeatVisitor.java
+++ b/heat/HeatVisitor.java
@@ -784,7 +784,7 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,ArrayList<TypeBundle>>
             t = new TypeInstance("null",null,null,null);
             ClassInstance c = this.recentClass;
             MinimalLogger.info(String.format("Setting class to the most recent return: %s",
-                                             c.toString()));
+                                             c));
             t.addClassInstance(c);
             break;
         default:
@@ -951,6 +951,7 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,ArrayList<TypeBundle>>
                                          n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
         TypeInstance t = this.symt.getType(n.f0.tokenImage);
+        this.recentClass = t.getClassInstance();
         _ret = new TypeBundle(t.getType(), t.getClassInstance());
         ///////////////////////////////////////////////////////////////
         MinimalLogger.info(String.format("<- %s with %s",
-- 
cgit v1.2.3