1 files changed, 1 insertions, 12 deletions
diff --git a/src/ir/module-utils.cpp b/src/ir/module-utils.cpp
index e0bd9592a..3c81d6679 100644
--- a/src/ir/module-utils.cpp
+++ b/src/ir/module-utils.cpp
@@ -506,9 +506,7 @@ InsertOrderedMap<HeapType, HeapTypeInfo> collectHeapTypeInfo(
   while (!newTypes.empty()) {
     while (!newTypes.empty()) {
       auto ht = newTypes.pop();
-      // TODO: Use getReferencedHeapTypes instead and remove separate
-      // consideration of supertypes below.
-      for (HeapType child : ht.getHeapTypeChildren()) {
+      for (HeapType child : ht.getReferencedHeapTypes()) {
         if (!child.isBasic()) {
           if (!info.contains(child)) {
             noteNewType(child);
@@ -517,15 +515,6 @@ InsertOrderedMap<HeapType, HeapTypeInfo> collectHeapTypeInfo(
         }
       }
 
-      if (auto super = ht.getDeclaredSuperType()) {
-        if (!info.contains(*super)) {
-          noteNewType(*super);
-          // TODO: This should be unconditional for the count to be correct, but
-          // this will be moot once we use getReferencedHeapTypes above.
-          info.note(*super);
-        }
-      }
-
       // Make sure we've noted the complete recursion group of each type as
       // well.
       if (inclusion != TypeInclusion::UsedIRTypes) {