Merge pull request #40 from drolevar/fix_revert_to_postbackendbind

Revert to using BaseBurnBackendBinderExtension::PostBackendBind

2 files changed, 4 insertions, 7 deletions

diff --git a/global.json b/global.json
index 9492f867..34e8e7e6 100644
--- a/global.json
+++ b/global.json
@@ -1,6 +1,6 @@
 {
   "msbuild-sdks": {
-    "WixToolset.Sdk": "4.0.0-build-0190"
+    "WixToolset.Sdk": "4.0.0-build-0192"
   },
   "sdk": {
     "allowPrerelease": false
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;