diff --git a/packages/page-agent/src/PageAgent.ts b/packages/page-agent/src/PageAgent.ts
index e6d0b4b..67cad4e 100644
--- a/packages/page-agent/src/PageAgent.ts
+++ b/packages/page-agent/src/PageAgent.ts
@@ -281,7 +281,7 @@ export class PageAgent extends EventTarget {
 				console.log(chalk.green('Step finished:'), actionName)
 				console.groupEnd()
 
-				await onAfterStep.call(this, step, this.history)
+				await onAfterStep.call(this, this.history)
 
 				step++
 				if (step > MAX_STEPS) {
diff --git a/packages/page-agent/src/config/index.ts b/packages/page-agent/src/config/index.ts
index 54e3b42..fc25dea 100644
--- a/packages/page-agent/src/config/index.ts
+++ b/packages/page-agent/src/config/index.ts
@@ -68,11 +68,7 @@ export interface AgentConfig {
 	// @todo: remove `this` binding, pass agent as explicit parameter instead
 
 	onBeforeStep?: (this: PageAgent, stepCnt: number) => Promise<void> | void
-	onAfterStep?: (
-		this: PageAgent,
-		stepCnt: number,
-		history: HistoricalEvent[]
-	) => Promise<void> | void
+	onAfterStep?: (this: PageAgent, history: HistoricalEvent[]) => Promise<void> | void
 	onBeforeTask?: (this: PageAgent) => Promise<void> | void
 	onAfterTask?: (this: PageAgent, result: ExecutionResult) => Promise<void> | void