diff --git a/doc/source/reference/update-provider-tree.rst b/doc/source/reference/update-provider-tree.rst
index 7ae102b25afa..2ffcbc7c6011 100644
--- a/doc/source/reference/update-provider-tree.rst
+++ b/doc/source/reference/update-provider-tree.rst
@@ -175,7 +175,7 @@ would become:
 
 .. code::
 
-  def update_provider_tree(self, provider_tree, nodename):
+  def update_provider_tree(self, provider_tree, nodename, allocations=None):
       inv_data = {
           'VCPU': { ... },
           'MEMORY_MB': { ... },
@@ -206,7 +206,7 @@ would become:
 
 .. code::
 
-  def update_provider_tree(self, provider_tree, nodename):
+  def update_provider_tree(self, provider_tree, nodename, allocations=None):
       provider_tree.add_traits(
           nodename, 'HW_CPU_X86_AVX', 'HW_CPU_X86_AVX2', 'CUSTOM_GOLD')