diff --git a/Scripts/AutomaticVMFSUnmap.ps1 b/Scripts/AutomaticVMFSUnmap.ps1
index fa240d9..2f32617 100755
--- a/Scripts/AutomaticVMFSUnmap.ps1
+++ b/Scripts/AutomaticVMFSUnmap.ps1
@@ -19,6 +19,8 @@ Function Get-VMFSUnmap {
         [VMware.VimAutomation.ViCore.Impl.V1.DatastoreManagement.DatastoreImpl]$Datastore
      )
 
+     $datastoreInfo = $Datastore.ExtensionData.Info
+
      if($datastoreInfo -is [VMware.Vim.VmfsDatastoreInfo] -and $datastoreInfo.Vmfs.MajorVersion -eq 6) {
         $datastoreInfo.Vmfs | select Name, UnmapPriority, UnmapGranularity
      } else {
@@ -65,4 +67,4 @@ Function Set-VMFSUnmap {
 
     Write-Host "$reconfigMessage ..."
     $storageSystem.UpdateVmfsUnmapPriority($uuid,$enableUNMAP)
-}
\ No newline at end of file
+}