diff --git a/src/Microsoft.AspNet.Razor/CodeGenerators/CSharpCodeGenerator.cs b/src/Microsoft.AspNet.Razor/CodeGenerators/CSharpCodeGenerator.cs
index e148b7b604..531965a754 100644
--- a/src/Microsoft.AspNet.Razor/CodeGenerators/CSharpCodeGenerator.cs
+++ b/src/Microsoft.AspNet.Razor/CodeGenerators/CSharpCodeGenerator.cs
@@ -24,8 +24,11 @@ namespace Microsoft.AspNet.Razor.CodeGenerators
private ChunkTree Tree { get { return Context.ChunkTreeBuilder.ChunkTree; } }
public RazorEngineHost Host { get { return Context.Host; } }
- // Internal for testing
- internal virtual CSharpCodeWriter CreateCodeWriter()
+ ///
+ /// Protected for testing.
+ ///
+ /// A new instance of .
+ protected virtual CSharpCodeWriter CreateCodeWriter()
{
return new CSharpCodeWriter();
}
diff --git a/test/Microsoft.AspNet.Razor.Test/CodeGenerators/CodeGenTestCodeGenerator.cs b/test/Microsoft.AspNet.Razor.Test/CodeGenerators/CodeGenTestCodeGenerator.cs
index 6457e40aad..be1a31bff2 100644
--- a/test/Microsoft.AspNet.Razor.Test/CodeGenerators/CodeGenTestCodeGenerator.cs
+++ b/test/Microsoft.AspNet.Razor.Test/CodeGenerators/CodeGenTestCodeGenerator.cs
@@ -12,7 +12,7 @@ namespace Microsoft.AspNet.Razor.Test.Generator
{
}
- internal override CSharpCodeWriter CreateCodeWriter()
+ protected override CSharpCodeWriter CreateCodeWriter()
{
return new TestCodeWriter();
}