diff --git a/docs/advanced_topics/custom_tasks.rst b/docs/advanced_topics/custom_tasks.rst
index d189f8e2b0..5b473f5619 100644
--- a/docs/advanced_topics/custom_tasks.rst
+++ b/docs/advanced_topics/custom_tasks.rst
@@ -105,6 +105,8 @@ Your custom task must then be instructed to generate an instance of your custom
 
         panels = Task.panels + [FieldPanel('user')]
 
+        task_state_class = UserApprovalTaskState
+
         # prevent editing of ``user`` after the task is created
         exclude_on_edit = {'user'}