diff --git a/docs/api.task_logic.rst b/docs/api.task_logic.rst
index 8e67ea5a..77c8be58 100644
--- a/docs/api.task_logic.rst
+++ b/docs/api.task_logic.rst
@@ -2,7 +2,7 @@ api.task_logic
 -------------
 
 
-.. automodule:: aind_behavior_vr_foraging.task_logic.AindVrForagingTaskLogic
+.. autopydantic_model:: aind_behavior_vr_foraging.task_logic.AindVrForagingTaskLogic
    :members:
    :undoc-members:
    :show-inheritance: