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;