diff --git a/plugin.go b/plugin.go
index cf986e0..7e3ab0a 100644
--- a/plugin.go
+++ b/plugin.go
@@ -226,9 +226,9 @@ func (p Plugin) Exec() error {
 	svc := lambda.New(sess, config)
 
 	if isUpdateConfig {
-		if err := svc.WaitUntilFunctionUpdatedWithContext(
+		if err := svc.WaitUntilFunctionUpdatedV2WithContext(
 			aws.BackgroundContext(),
-			&lambda.GetFunctionConfigurationInput{
+			&lambda.GetFunctionInput{
 				FunctionName: aws.String(p.Config.FunctionName),
 			},
 			request.WithWaiterMaxAttempts(p.Config.MaxAttempts),