Use SymbolsFinalized instead of BundleFinalize

1 files changed, 3 insertions, 6 deletions

diff --git a/src/wixext/BalBurnBackendExtension.cs b/src/wixext/BalBurnBackendExtension.cs
index bf96e388..5e647b1c 100644
--- a/src/wixext/BalBurnBackendExtension.cs
+++ b/src/wixext/BalBurnBackendExtension.cs
@@ -11,7 +11,7 @@ namespace WixToolset.Bal
     using WixToolset.Data.Symbols;
     using WixToolset.Extensibility;
 
-    public class BalBurnBackendExtension : BaseBurnBackendExtension
+    public class BalBurnBackendExtension : BaseBurnBackendBinderExtension
     {
         private static readonly IntermediateSymbolDefinition[] BurnSymbolDefinitions =
         {
@@ -27,12 +27,9 @@ namespace WixToolset.Bal
 
         protected override IEnumerable<IntermediateSymbolDefinition> SymbolDefinitions => BurnSymbolDefinitions;
 
-        public override void BundleFinalize()
+        public override void SymbolsFinalized(IntermediateSection section)
         {
-            base.BundleFinalize();
-
-            var intermediate = this.Context.IntermediateRepresentation;
-            var section = intermediate.Sections.Single();
+            base.SymbolsFinalized(section);
 
             var baSymbol = section.Symbols.OfType<WixBootstrapperApplicationDllSymbol>().SingleOrDefault();
             var baId = baSymbol?.Id?.Id;