diff --git a/rivetkit-typescript/packages/rivetkit/src/workflow/context.ts b/rivetkit-typescript/packages/rivetkit/src/workflow/context.ts
index 9c148e8f96..2938f524a4 100644
--- a/rivetkit-typescript/packages/rivetkit/src/workflow/context.ts
+++ b/rivetkit-typescript/packages/rivetkit/src/workflow/context.ts
@@ -239,7 +239,7 @@ export class ActorWorkflowContext<
 	}
 
 	async step<T>(
-		nameOrConfig: string | Parameters<WorkflowContextInterface["step"]>[0],
+		nameOrConfig: string | StepConfig<T>,
 		run?: () => Promise<T>,
 	): Promise<T> {
 		if (typeof nameOrConfig === "string") {
@@ -261,9 +261,7 @@ export class ActorWorkflowContext<
 	}
 
 	async tryStep<T>(
-		nameOrConfig:
-			| string
-			| Parameters<WorkflowContextInterface["tryStep"]>[0],
+		nameOrConfig: string | TryStepConfig<T>,
 		run?: () => Promise<T>,
 	): Promise<TryStepResult<T>> {
 		if (typeof nameOrConfig === "string") {