diff --git a/tests/workbenches/docs.py b/tests/workbenches/docs.py
index d84c9643..5fcab3e4 100644
--- a/tests/workbenches/docs.py
+++ b/tests/workbenches/docs.py
@@ -1,32 +1,38 @@
-def Desc(value: str):
+from typing import Callable
+import typing
+
+T = typing.TypeVar("T")
+
+
+def Desc(value: str) -> str:
     return value
 
 
 def Step(
-    value: str,
-    expected: str,
-):
+        value: str,
+        expected: str,
+) -> tuple[str, str]:
     return value, expected
 
 
 def SuiteDoc(
-    description: str,
-    beforeTestSteps: set[Step],
-    afterTestSteps: set[Step],
-):
+        description: str,
+        beforeTestSteps: set[Step],
+        afterTestSteps: set[Step],
+) -> Callable[[T], T]:
     return lambda x: x
 
 
 def Contact(
-    name: str,
-    email: str,
-):
+        name: str,
+        email: str,
+) -> tuple[str, str]:
     return name, email
 
 
 def TestDoc(
-    description: str,
-    contact: str,
-    steps: set[Step],
-):
+        description: str,
+        contact: str,
+        steps: set[Step],
+) -> Callable[[T], T]:
     return lambda x: x