diff --git a/packages/arb-token-bridge-ui/src/components/TransferPanel/TransferPanel.tsx b/packages/arb-token-bridge-ui/src/components/TransferPanel/TransferPanel.tsx
index 4e07d5e7fb..3e1dd3f923 100644
--- a/packages/arb-token-bridge-ui/src/components/TransferPanel/TransferPanel.tsx
+++ b/packages/arb-token-bridge-ui/src/components/TransferPanel/TransferPanel.tsx
@@ -383,7 +383,7 @@ export function TransferPanel() {
     return true
   }
 
-  const stepExecutor: UiDriverStepExecutor = async step => {
+  const stepExecutor: UiDriverStepExecutor = async (context, step) => {
     if (process.env.NODE_ENV === 'development') {
       console.log(step)
     }
diff --git a/packages/arb-token-bridge-ui/src/ui-driver/UiDriver.test.ts b/packages/arb-token-bridge-ui/src/ui-driver/UiDriver.test.ts
index c25d8cb676..6ced220125 100644
--- a/packages/arb-token-bridge-ui/src/ui-driver/UiDriver.test.ts
+++ b/packages/arb-token-bridge-ui/src/ui-driver/UiDriver.test.ts
@@ -16,7 +16,7 @@ it('successfully does step execution', async () => {
     yield { type: 'start' }
   }
 
-  const executor: UiDriverStepExecutor = async function (step) {
+  const executor: UiDriverStepExecutor = async function (_, step) {
     if (step.type === 'start') {
       counter += 1
     }
@@ -39,7 +39,7 @@ it('successfully does early return', async () => {
     yield { type: 'start' }
   }
 
-  const executor: UiDriverStepExecutor = async function (step) {
+  const executor: UiDriverStepExecutor = async function (_, step) {
     if (step.type === 'start') {
       counter += 1
     }
diff --git a/packages/arb-token-bridge-ui/src/ui-driver/UiDriver.ts b/packages/arb-token-bridge-ui/src/ui-driver/UiDriver.ts
index e8de84464a..6cc929cfd8 100644
--- a/packages/arb-token-bridge-ui/src/ui-driver/UiDriver.ts
+++ b/packages/arb-token-bridge-ui/src/ui-driver/UiDriver.ts
@@ -64,6 +64,7 @@ export type UiDriverStepGenerator<TStep extends UiDriverStep = UiDriverStep> = (
 ) => AsyncGenerator<TStep, void, UiDriverStepResultFor<TStep['type']>>
 
 export type UiDriverStepExecutor<TStep extends UiDriverStep = UiDriverStep> = (
+  context: UiDriverContext,
   step: TStep
 ) => Promise<UiDriverStepResultFor<TStep['type']>>
 
@@ -100,7 +101,7 @@ export async function drive<TStep extends UiDriverStep>(
     }
 
     // execute current step and obtain the result
-    const result = await executor(step)
+    const result = await executor(context, step)
 
     // pass the result back into the generator
     nextStep = await flow.next(result)