From de08f0eece205cc2e0dd093817882aa0b760b30d Mon Sep 17 00:00:00 2001
From: Wes Carroll <wcarroll@users.noreply.github.com>
Date: Thu, 19 Mar 2020 17:08:08 -0400
Subject: [PATCH] Ensure the right DSId is mapped to the target host.

---
 ZertoApiWrapper/Public/Install-ZertoVra.ps1 | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/ZertoApiWrapper/Public/Install-ZertoVra.ps1 b/ZertoApiWrapper/Public/Install-ZertoVra.ps1
index 2bca65f..1199117 100644
--- a/ZertoApiWrapper/Public/Install-ZertoVra.ps1
+++ b/ZertoApiWrapper/Public/Install-ZertoVra.ps1
@@ -40,6 +40,14 @@ function Install-ZertoVra {
         $hostIdentifier = Get-ZertoVirtualizationSite -siteIdentifier $siteIdentifier -hosts | Where-Object {$_.VirtualizationHostName -eq $hostName} | Select-Object hostIdentifier -ExpandProperty hostIdentifier
         $networkIdentifier = Get-ZertoVirtualizationSite -siteIdentifier $siteIdentifier -networks | Where-Object {$_.VirtualizationNetworkName -eq $networkName} | Select-Object NetworkIdentifier -ExpandProperty NetworkIdentifier
         $datastoreIdentifier = Get-ZertoVirtualizationSite -siteIdentifier $siteIdentifier -datastores | Where-Object {$_.DatastoreName -eq $datastoreName} | Select-Object DatastoreIdentifier -ExpandProperty DatastoreIdentifier
+        if ($datastoreIdentifier.count -gt 1){
+            $hostDevices = Get-ZertoVirtualizationSite -siteIdentifier $siteIdentifier -devices -hostIdentifier $hostIdentifier
+            $datastoreIdentifier = foreach ($identifier in $datastoreIdentifier) {
+                if ($identifier -in $hostDevices.DatastoreIdentifier){
+                    $identifier
+                }
+            }
+        }
 
         # Build the JSON object through an Ordered Hashtable.
         $vraBasic = [ordered]@{}