diff --git a/setup.py b/setup.py
index 7b4bc6d36..531cfc7c2 100644
--- a/setup.py
+++ b/setup.py
@@ -249,7 +249,7 @@ def run(self):
     ],
     extras_require={
         'dev': [
-            'azure-functions==1.0.3',
+            'azure-functions==1.0.4',
             'flake8~=3.5.0',
             'mypy',
             'pytest',
diff --git a/tests/endtoend/servicebus_functions/servicebus_trigger/__init__.py b/tests/endtoend/servicebus_functions/servicebus_trigger/__init__.py
index 46e70c459..b70905930 100644
--- a/tests/endtoend/servicebus_functions/servicebus_trigger/__init__.py
+++ b/tests/endtoend/servicebus_functions/servicebus_trigger/__init__.py
@@ -8,6 +8,7 @@ def main(msg: azf.ServiceBusMessage) -> str:
         'message_id': msg.message_id,
         'body': msg.get_body().decode('utf-8'),
         'content_type': msg.content_type,
+        'delivery_count': msg.delivery_count,
         'expiration_time': msg.expiration_time,
         'label': msg.label,
         'partition_key': msg.partition_key,