Added a try catch to first WMI call in the 'ByComputerName' parameter set. If the call fails the entire loop for that system will be terminated.

Author: dotps1

Functions/Test-WannaCryVulnerability.ps1

@@ -1,7 +1,7 @@
 <#PSScriptInfo
 
 .Version
-    1.2
+    1.3
 .Guid
     477aa3f4-0434-4925-9c92-7323066cceb7
 .Author 
@@ -55,6 +55,8 @@
     SMB1ProtocolEnabled : True
 .Notes
     Not applicable to windows 10.
+.Link
+    https://www.redsocks.eu/news/ransomware-wannacry/
 .Link
     https://dotps1.github.io
 .Link
@@ -158,7 +160,7 @@ process {
                     continue
                 }
 
-                if ($osCaption -match "Windows 10|2016") {
+                if ($osCaption -match "Windows 10") {
                     Write-Warning -Message "$osCaption is not vulnerable to WannaCry."
                     continue
                 }
@@ -212,7 +214,7 @@ process {
                 $osCaption = Get-CimInstance -CimSession $cimSessionValue -ClassName Win32_OperatingSystem -Property Caption | 
                     Select-Object -ExpandProperty Caption
 
-                if ($osCaption -match "Windows 10|2016") {
+                if ($osCaption -match "Windows 10") {
                     Write-Error -Message "$osCaption is not vulnerable to WannaCry."
                     continue
                 }