diff --git a/src/Scripts/Register.ps1 b/src/Scripts/Register.ps1
index b5a7813..03751db 100644
--- a/src/Scripts/Register.ps1
+++ b/src/Scripts/Register.ps1
@@ -31,10 +31,10 @@ $VerbosePreference = "Continue"
 $ProgressPreference = "SilentlyContinue"
 
 $walAssemblyName = "MicrosoftServices.IdentityManagement.WorkflowActivityLibrary"
-$walAssemblyPath = Join-Path $PWD -ChildPath "$walAssemblyName.dll"
-$walUIAssemblyPath = Join-Path $PWD -ChildPath "$walAssemblyName.UI.dll"
-$gacUtilExePath = Join-Path $PWD -ChildPath "gacutil.exe"
-$snExePath = Join-Path $PWD -ChildPath "sn.exe"
+$walAssemblyPath = Join-Path $PWD.ProviderPath -ChildPath "$walAssemblyName.dll"
+$walUIAssemblyPath = Join-Path $PWD.ProviderPath -ChildPath "$walAssemblyName.UI.dll"
+$gacUtilExePath = Join-Path $PWD.ProviderPath -ChildPath "gacutil.exe"
+$snExePath = Join-Path $PWD.ProviderPath -ChildPath "sn.exe"
 
 function TestIsAssemblyLoaded
 {
diff --git a/src/Scripts/UpdateWorkflowXoml.ps1 b/src/Scripts/UpdateWorkflowXoml.ps1
index be86a7a..52532ea 100644
Binary files a/src/Scripts/UpdateWorkflowXoml.ps1 and b/src/Scripts/UpdateWorkflowXoml.ps1 differ