diff --git a/aws_lambda/aws_lambda.py b/aws_lambda/aws_lambda.py
index 780715a9..e02bbf66 100644
--- a/aws_lambda/aws_lambda.py
+++ b/aws_lambda/aws_lambda.py
@@ -16,7 +16,6 @@
 import boto3
 import botocore
 import yaml
-import sys
 
 from .helpers import archive
 from .helpers import get_environment_variable_value
@@ -703,6 +702,10 @@ def update_function(
             ZipFile=byte_stream,
             Publish=True,
         )
+    waiter = client.get_waiter('function_updated')
+    waiter.wait(
+        FunctionName=cfg.get("function_name"),
+    )
 
     kwargs = {
         "FunctionName": cfg.get("function_name"),