From 670f8e6b192b44147588c944ebc71b84333fd87b Mon Sep 17 00:00:00 2001
From: Kashif Saadat <kashifsaadat@gmail.com>
Date: Mon, 5 Feb 2018 16:30:45 +0000
Subject: [PATCH] Fix drain command for rolling-updates

---
 pkg/instancegroups/instancegroups.go | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/pkg/instancegroups/instancegroups.go b/pkg/instancegroups/instancegroups.go
index 5ba30e6473..b5edfa4ab2 100644
--- a/pkg/instancegroups/instancegroups.go
+++ b/pkg/instancegroups/instancegroups.go
@@ -24,7 +24,6 @@ import (
 	"time"
 
 	"github.com/golang/glog"
-	"github.com/spf13/cobra"
 	api "k8s.io/kops/pkg/apis/kops"
 	"k8s.io/kops/pkg/cloudinstances"
 	"k8s.io/kops/pkg/featureflag"
@@ -313,9 +312,7 @@ func (r *RollingUpdateInstanceGroup) DrainNode(u *cloudinstances.CloudInstanceGr
 		ErrOut:           errOut,
 	}
 
-	cmd := &cobra.Command{
-		Use: "cordon NODE",
-	}
+	cmd := cmd.NewCmdDrain(f, out, errOut)
 	args := []string{u.Node.Name}
 	err := options.SetupDrain(cmd, args)
 	if err != nil {