From baf9b1b4d284ce7295ee0b36d67dabc51c958c08 Mon Sep 17 00:00:00 2001
From: sethvs <sethvs@users.noreply.github.com>
Date: Tue, 27 Feb 2024 05:34:33 +0300
Subject: [PATCH] Update completers

---
 functions/DiskSmartInfo.completers.ps1 | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/functions/DiskSmartInfo.completers.ps1 b/functions/DiskSmartInfo.completers.ps1
index ded4394..3eca93b 100644
--- a/functions/DiskSmartInfo.completers.ps1
+++ b/functions/DiskSmartInfo.completers.ps1
@@ -114,10 +114,18 @@ class DiskCompleter : IArgumentCompleter
             {
                 $sessionParameters.Add('Credential', $fakeBoundParameters.Credential)
             }
-            if ($cimSession = New-CimSession @sessionParameters)
+
+            $option = [Microsoft.Management.Infrastructure.Options.WSManSessionOptions]::new()
+            $option.Timeout = New-TimeSpan -Seconds 1
+
+            if ($cimSession = New-CimSession -SessionOption $option @sessionParameters)
             {
                 $instanceParameters.Add('CimSession', $cimSession)
             }
+            else
+            {
+                return $result
+            }
         }
         elseif  ($fakeBoundParameters.ContainsKey('CimSession'))
         {