diff --git a/examples/logger/package-lock.json b/examples/logger/package-lock.json
index 40ce86d..c74dda3 100644
--- a/examples/logger/package-lock.json
+++ b/examples/logger/package-lock.json
@@ -25,7 +25,7 @@
     },
     "../..": {
       "name": "@logtail/next",
-      "version": "0.2.0",
+      "version": "0.2.1",
       "license": "MIT",
       "dependencies": {
         "use-deep-compare": "^1.2.1",
diff --git a/src/withBetterStack.ts b/src/withBetterStack.ts
index f8b6bce..094db1f 100644
--- a/src/withBetterStack.ts
+++ b/src/withBetterStack.ts
@@ -49,10 +49,14 @@ export function withBetterStackNextConfig(nextConfig: NextConfig): NextConfig {
 }
 
 export type BetterStackRequest = NextRequest & { log: Logger };
-type NextHandler<T = any> = (
+type BetterStackRouteHandler<T = any> = (
   req: BetterStackRequest,
   arg?: T
 ) => Promise<Response> | Promise<NextResponse> | NextResponse | Response;
+type NextHandler<T = any> = (
+  req: NextRequest,
+  arg?: T
+) => Promise<Response> | Promise<NextResponse> | NextResponse | Response;
 
 type BetterStackRouteHandlerConfig = {
   logRequestDetails?: boolean | (keyof RequestJSON)[];
@@ -61,7 +65,10 @@ type BetterStackRouteHandlerConfig = {
   redirectLogLevel?: LogLevel; // defaults to LogLevel.info
 };
 
-export function withBetterStackRouteHandler(handler: NextHandler, config?: BetterStackRouteHandlerConfig): NextHandler {
+export function withBetterStackRouteHandler(
+  handler: BetterStackRouteHandler,
+  config?: BetterStackRouteHandlerConfig
+): NextHandler {
   return async (req: Request | NextRequest, arg: any) => {
     let region = '';
     if ('geo' in req) {
@@ -183,7 +190,7 @@ export function withBetterStackRouteHandler(handler: NextHandler, config?: Bette
   };
 }
 
-type WithBetterStackParam = NextConfig | NextHandler;
+type WithBetterStackParam = NextConfig | BetterStackRouteHandler;
 
 function isNextConfig(param: WithBetterStackParam): param is NextConfig {
   return typeof param == 'object';
@@ -191,7 +198,7 @@ function isNextConfig(param: WithBetterStackParam): param is NextConfig {
 
 // withBetterStack can be called either with NextConfig, which will add proxy rewrites
 // to improve deliverability of Web-Vitals and logs.
-export function withBetterStack(param: NextHandler, config?: BetterStackRouteHandlerConfig): NextHandler;
+export function withBetterStack(param: BetterStackRouteHandler, config?: BetterStackRouteHandlerConfig): NextHandler;
 export function withBetterStack(param: NextConfig): NextConfig;
 export function withBetterStack(param: WithBetterStackParam, config?: BetterStackRouteHandlerConfig) {
   if (typeof param == 'function') {