diff --git a/packages/doorman-ui/src/routers/QueryRouter.tsx b/packages/doorman-ui/src/routers/QueryRouter.tsx
index 3c5198f..9264ed8 100644
--- a/packages/doorman-ui/src/routers/QueryRouter.tsx
+++ b/packages/doorman-ui/src/routers/QueryRouter.tsx
@@ -1,14 +1,14 @@
-import React from "react";
+import React, { ReactElement } from "react";
 import { ReactNode } from "react";
 import { useLocation, useSearchParams } from "react-router-dom";
 
 export interface IQueryRouterProps {
-  mapping: Record<string, ReactNode>;
+  mapping: Record<string, ReactElement>;
 };
 
-export const QueryRouter = ({ mapping }: IQueryRouterProps) => {
+export const QueryRouter = ({ mapping }: IQueryRouterProps): ReactElement => {
   const [params] = useSearchParams();
-  let element = null;
+  let element: ReactElement | null = null;
 
   for (const key of params.keys()) {
     if (mapping[key]) {
@@ -16,6 +16,7 @@ export const QueryRouter = ({ mapping }: IQueryRouterProps) => {
       break;
     }
   }
+
   if (element === null) {
     throw new Error("missing mapping");
   }