diff --git a/src/RazorPageGenerator/Program.cs b/src/RazorPageGenerator/Program.cs
index f3ae757886..49a3154c90 100644
--- a/src/RazorPageGenerator/Program.cs
+++ b/src/RazorPageGenerator/Program.cs
@@ -7,6 +7,7 @@ using System.IO;
 using System.Linq;
 using System.Text;
 using Microsoft.AspNetCore.Razor.Language;
+using Microsoft.AspNetCore.Razor.Language.Extensions;
 
 namespace RazorPageGenerator
 {
@@ -60,6 +61,10 @@ Examples:
                         @class.Modifiers.Add("internal");
                     });
 
+                FunctionsDirective.Register(builder);
+                InheritsDirective.Register(builder);
+                SectionDirective.Register(builder);
+
                 builder.Features.Add(new SuppressChecksumOptionsFeature());
                 builder.Features.Add(new SuppressMetadataAttributesFeature());