From ae6ac6e148e895e58c17bddf051768a41669ddf2 Mon Sep 17 00:00:00 2001
From: John Hodge <tpg@mutabah.net>
Date: Fri, 18 Nov 2016 13:46:47 +0800
Subject: HIR Expand ErasedTypes - Visit result types

---
 src/hir_expand/erased_types.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src')

diff --git a/src/hir_expand/erased_types.cpp b/src/hir_expand/erased_types.cpp
index 4da148d0..93bef966 100644
--- a/src/hir_expand/erased_types.cpp
+++ b/src/hir_expand/erased_types.cpp
@@ -28,12 +28,19 @@ namespace {
         void visit_root(::HIR::ExprPtr& root)
         {
             root->visit(*this);
+            visit_type(root->m_res_type);
             for(auto& ty : root.m_bindings)
                 visit_type(ty);
             for(auto& ty : root.m_erased_types)
                 visit_type(ty);
         }
         
+        void visit_node_ptr(::std::unique_ptr< ::HIR::ExprNode>& node_ptr) override {
+            assert(node_ptr);
+            node_ptr->visit(*this);
+            visit_type(node_ptr->m_res_type);
+        }
+        
         void visit_type(::HIR::TypeRef& ty) override
         {
             static Span sp;
-- 
cgit v1.2.3