From dd1ae2fc671bc50870b121322346851b54bc5c86 Mon Sep 17 00:00:00 2001
From: youyo <1003ni2@gmail.com>
Date: Tue, 30 Nov 2021 02:31:09 +0900
Subject: [PATCH] bugfix

---
 cli.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/cli.go b/cli.go
index 6c22433..e359589 100644
--- a/cli.go
+++ b/cli.go
@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"regexp"
 	"strings"
+	"time"
 
 	"github.com/k1LoW/duration"
 	"github.com/manifoldco/promptui"
@@ -75,12 +76,12 @@ func Run(cmd *cobra.Command, args []string) (err error) {
 	}
 	defer cmdPortForwarding.Process.Kill()
 
-	waitOpenPort(ConnectHost, localPort)
-
 	if err = sendSSHPublicKey(ctx, awsSession, instanceID, viper.GetString("username"), viper.GetString("publicKey")); err != nil {
 		return err
 	}
 
+	time.Sleep(1 * time.Second)
+
 	cmdSsh, err := execSshCommand(ctx, viper.GetString("username"), ConnectHost, localPort, viper.GetString("identity-file"))
 	cmdSsh.Wait()