HIR Expand ErasedTypes - Visit result types

1 files changed, 7 insertions, 0 deletions

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;