diff --git a/src/middleware/promptInjectionGuard.ts b/src/middleware/promptInjectionGuard.ts
index d257013278..be2e149479 100644
--- a/src/middleware/promptInjectionGuard.ts
+++ b/src/middleware/promptInjectionGuard.ts
@@ -52,7 +52,7 @@ export function createInjectionGuard(options: PromptInjectionGuardrailOptions =
 export function withInjectionGuard(handler: any, options: any = {}) {
   const guard = createInjectionGuard(options);
 
-  return async function guardedHandler(request: any, context: any) {
+  return async function guardedHandler(request: any, context?: any) {
     // Only apply to POST/PUT/PATCH
     if (!["POST", "PUT", "PATCH"].includes(request.method)) {
       return handler(request, context);