Fix loading BAFunctions.

1 files changed, 4 insertions, 2 deletions

diff --git a/src/ext/Bal/wixext/BalBurnBackendExtension.cs b/src/ext/Bal/wixext/BalBurnBackendExtension.cs
index e8dc7a3e..c6a1e0c1 100644
--- a/src/ext/Bal/wixext/BalBurnBackendExtension.cs
+++ b/src/ext/Bal/wixext/BalBurnBackendExtension.cs
@@ -107,12 +107,14 @@ namespace WixToolset.Bal
             }
             else
             {
-                var payloadId = baFunctionsSymbol.Id;
-                var bundlePayloadSymbol = payloadPropertiesSymbols.Single(x => payloadId == x.Id);
+                var payloadId = baFunctionsSymbol.PayloadId;
+                var bundlePayloadSymbol = payloadPropertiesSymbols.Single(x => payloadId == x.Id.Id);
                 if (BurnConstants.BurnUXContainerName != bundlePayloadSymbol.ContainerRef)
                 {
                     this.Messaging.Write(BalErrors.BAFunctionsPayloadRequiredInUXContainer(baFunctionsSymbol.SourceLineNumbers));
                 }
+
+                baFunctionsSymbol.FilePath = bundlePayloadSymbol.Name;
             }
         }