diff --git a/aws_lambda/aws_lambda.py b/aws_lambda/aws_lambda.py
index 780715a9..0b5ca884 100644
--- a/aws_lambda/aws_lambda.py
+++ b/aws_lambda/aws_lambda.py
@@ -704,6 +704,10 @@ def update_function(
             Publish=True,
         )
 
+    # Wait for function to be updated
+    waiter = client.get_waiter('function_updated')
+    waiter.wait(FunctionName=cfg.get("function_name"))
+
     kwargs = {
         "FunctionName": cfg.get("function_name"),
         "Role": role,
diff --git a/setup.py b/setup.py
index 5f6ca2f8..bce3297e 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
 
 REQUIREMENTS = [
     "boto3>=1.4.4",
-    "click==6.6",
+    "click>=6.6",
     "PyYAML==5.1",
 ]
 PACKAGE_DATA = {