diff --git a/skyvern/webeye/actions/actions.py b/skyvern/webeye/actions/actions.py
index 61b8de08..3aaa16ff 100644
--- a/skyvern/webeye/actions/actions.py
+++ b/skyvern/webeye/actions/actions.py
@@ -242,7 +242,7 @@ class CompleteAction(DecisiveAction):
 class ExtractAction(Action):
     action_type: ActionType = ActionType.EXTRACT
     data_extraction_goal: str | None = None
-    data_extraction_schema: dict[str, Any] | None = None
+    data_extraction_schema: dict[str, Any] | list | str | None = None
 
 
 class ScrollAction(Action):