From 744da76e7ccfc906a2a421b2c08b746c09566340 Mon Sep 17 00:00:00 2001
From: Wataru Ashihara <wataash0607@gmail.com>
Date: Mon, 22 Oct 2018 10:31:16 +0900
Subject: [PATCH] Fix type

---
 main.go   | 4 ++--
 plugin.go | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index ac19934..803f3ae 100644
--- a/main.go
+++ b/main.go
@@ -81,7 +81,7 @@ func main() {
 			Usage:  "connection timeout",
 			EnvVar: "PLUGIN_TIMEOUT,SSH_TIMEOUT",
 		},
-		cli.IntFlag{
+		cli.DurationFlag{
 			Name:   "command.timeout,T",
 			Usage:  "command timeout",
 			EnvVar: "PLUGIN_COMMAND_TIMEOUT,SSH_COMMAND_TIMEOUT",
@@ -200,7 +200,7 @@ func run(c *cli.Context) error {
 			Host:           c.StringSlice("host"),
 			Port:           c.Int("port"),
 			Timeout:        c.Duration("timeout"),
-			CommandTimeout: c.Int("command.timeout"),
+			CommandTimeout: c.Duration("command.timeout"),
 			Script:         c.StringSlice("script"),
 			ScriptStop:     c.Bool("script.stop"),
 			Secrets:        c.StringSlice("secrets"),
diff --git a/plugin.go b/plugin.go
index e194772..f2a8bea 100644
--- a/plugin.go
+++ b/plugin.go
@@ -29,7 +29,7 @@ type (
 		Host           []string
 		Port           int
 		Timeout        time.Duration
-		CommandTimeout int
+		CommandTimeout time.Duration
 		Script         []string
 		ScriptStop     bool
 		Secrets        []string