Minor cleanup

WildernessLabs · Mar 6, 2024 · ea406a8 · ea406a8
1 parent e2833ac
commit ea406a8
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/Source/v2/Meadow.Cli/Commands/Current/Firmware/FirmwareWriteCommand.cs b/Source/v2/Meadow.Cli/Commands/Current/Firmware/FirmwareWriteCommand.cs
@@ -248,10 +248,7 @@ protected override async ValueTask ExecuteCommand()
 
     private async Task<IMeadowConnection?> WriteRuntime(IMeadowConnection? connection, DeviceInfo? deviceInfo, string runtimePath, string destinationFilename)
     {
-        if (connection == null)
-        {
-            connection = await GetConnectionAndDisableRuntime();
-        }
+        connection ??= await GetConnectionAndDisableRuntime();
 
         Logger?.LogInformation($"{Environment.NewLine}Writing Runtime ...");
 
@@ -280,6 +277,8 @@ protected override async ValueTask ExecuteCommand()
                     throw CommandException.MeadowDeviceNotFound;
                 }
 
+                connection = await GetCurrentConnection(true);
+
                 Logger?.LogInformation($"Meadow found at {newPort}");
 
                 // configure the route to that port for the user
@@ -318,10 +317,7 @@ private async Task WriteEspFiles(IMeadowConnection? connection, DeviceInfo? devi
             };
         }
 
-        if (deviceInfo == null)
-        {
-            deviceInfo = await connection.GetDeviceInfo(CancellationToken);
-        }
+        deviceInfo ??= await connection.GetDeviceInfo(CancellationToken);
 
         if (UseDfu || RequiresDfuForEspUpdates(deviceInfo))
         {