From 5a7d9b3fe5167532e9d7b6dfc861d430622d8819 Mon Sep 17 00:00:00 2001
From: triton <ripzonetriton@gmail.com>
Date: Tue, 26 Feb 2013 23:43:26 +0000
Subject: [PATCH] Use the new type reference class in unit in the CLI
 generator.

---
 src/Generator/Generators/CLI/CLISourcesTemplate.cs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/Generator/Generators/CLI/CLISourcesTemplate.cs b/src/Generator/Generators/CLI/CLISourcesTemplate.cs
index 569678d2..bdb6d6eb 100644
--- a/src/Generator/Generators/CLI/CLISourcesTemplate.cs
+++ b/src/Generator/Generators/CLI/CLISourcesTemplate.cs
@@ -37,8 +37,10 @@ namespace Cxxi.Generators.CLI
         {
             var includes = new HashSet<string>();
 
+            var typeRefs = unit.TypeReferences as TypeRefsVisitor;
+
             // Generate the forward references.
-            foreach (var forwardRef in unit.ForwardReferences)
+            foreach (var forwardRef in typeRefs.ForwardReferences)
             {
                 var decl = forwardRef;