linux2Java
diff --git a/‎ecs-cli/modules/aws/clients/cloudformation/client.go‎
Lines changed: 2 additions & 1 deletion b/‎ecs-cli/modules/aws/clients/cloudformation/client.go‎
Lines changed: 2 additions & 1 deletion
diff --git a/‎ecs-cli/modules/aws/clients/cloudformation/mock/sdk/cloudformationiface_mock.go‎
Lines changed: 22 additions & 0 deletions b/‎ecs-cli/modules/aws/clients/cloudformation/mock/sdk/cloudformationiface_mock.go‎
Lines changed: 22 additions & 0 deletions
diff --git a/‎ecs-cli/modules/aws/clients/ec2/client.go‎
Lines changed: 2 additions & 1 deletion b/‎ecs-cli/modules/aws/clients/ec2/client.go‎
Lines changed: 2 additions & 1 deletion
@@ -24,6 +24,7 @@ import (
 	"github.com/aws/amazon-ecs-cli/ecs-cli/utils"
 	"github.com/aws/aws-sdk-go/aws"
 	"github.com/aws/aws-sdk-go/aws/awserr"
+	"github.com/aws/aws-sdk-go/aws/session"
 	"github.com/aws/aws-sdk-go/service/cloudformation"
 	"github.com/aws/aws-sdk-go/service/cloudformation/cloudformationiface"
 )
@@ -116,7 +117,7 @@ func NewCloudformationClient() CloudformationClient {
 
 // Initialize initializes all the fields of the cloudFormationClient object.
 func (c *cloudformationClient) Initialize(params *config.CliParams) {
-	cfnClient := cloudformation.New(params.Config)
+	cfnClient := cloudformation.New(session.New(params.Config))
 	cfnClient.Handlers.Build.PushBackNamed(clients.CustomUserAgentHandler())
 	c.client = cfnClient
 	c.cliParams = params
 
@@ -65,6 +65,28 @@ func (_mr *_MockCloudFormationAPIRecorder) CancelUpdateStackRequest(arg0 interfa
 	return _mr.mock.ctrl.RecordCall(_mr.mock, "CancelUpdateStackRequest", arg0)
 }
 
+func (_m *MockCloudFormationAPI) ContinueUpdateRollback(_param0 *cloudformation.ContinueUpdateRollbackInput) (*cloudformation.ContinueUpdateRollbackOutput, error) {
+	ret := _m.ctrl.Call(_m, "ContinueUpdateRollback", _param0)
+	ret0, _ := ret[0].(*cloudformation.ContinueUpdateRollbackOutput)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+func (_mr *_MockCloudFormationAPIRecorder) ContinueUpdateRollback(arg0 interface{}) *gomock.Call {
+	return _mr.mock.ctrl.RecordCall(_mr.mock, "ContinueUpdateRollback", arg0)
+}
+
+func (_m *MockCloudFormationAPI) ContinueUpdateRollbackRequest(_param0 *cloudformation.ContinueUpdateRollbackInput) (*request.Request, *cloudformation.ContinueUpdateRollbackOutput) {
+	ret := _m.ctrl.Call(_m, "ContinueUpdateRollbackRequest", _param0)
+	ret0, _ := ret[0].(*request.Request)
+	ret1, _ := ret[1].(*cloudformation.ContinueUpdateRollbackOutput)
+	return ret0, ret1
+}
+
+func (_mr *_MockCloudFormationAPIRecorder) ContinueUpdateRollbackRequest(arg0 interface{}) *gomock.Call {
+	return _mr.mock.ctrl.RecordCall(_mr.mock, "ContinueUpdateRollbackRequest", arg0)
+}
+
 func (_m *MockCloudFormationAPI) CreateStack(_param0 *cloudformation.CreateStackInput) (*cloudformation.CreateStackOutput, error) {
 	ret := _m.ctrl.Call(_m, "CreateStack", _param0)
 	ret0, _ := ret[0].(*cloudformation.CreateStackOutput)
 
@@ -19,6 +19,7 @@ import (
 	"github.com/aws/amazon-ecs-cli/ecs-cli/modules/aws/clients"
 	"github.com/aws/amazon-ecs-cli/ecs-cli/modules/config"
 	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/session"
 	"github.com/aws/aws-sdk-go/service/ec2"
 	"github.com/aws/aws-sdk-go/service/ec2/ec2iface"
 )
@@ -38,7 +39,7 @@ type ec2Client struct {
 
 // NewEC2Client creates an instance of ec2Client object.
 func NewEC2Client(params *config.CliParams) EC2Client {
-	client := ec2.New(params.Config)
+	client := ec2.New(session.New(params.Config))
 	client.Handlers.Build.PushBackNamed(clients.CustomUserAgentHandler())
 	return &ec2Client{
 		client: client,