Defer ssh.Close() to always close SSH connection in prepare (#100)

cirruslabs · Jan 22, 2025 · 9211227 · 9211227
1 parent 7973405
commit 9211227
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/internal/commands/prepare/prepare.go b/internal/commands/prepare/prepare.go
@@ -140,6 +140,7 @@ func runPrepareVM(cmd *cobra.Command, args []string) error {
 	if err != nil {
 		return err
 	}
+	defer ssh.Close()
 
 	log.Println("Was able to SSH!")
 
@@ -251,7 +252,7 @@ func runPrepareVM(cmd *cobra.Command, args []string) error {
 
 	log.Println("VM is ready.")
 
-	return ssh.Close()
+	return nil
 }
 
 func ensureImageIsAllowed(image string) error {