diff --git a/app/play/[level]/test-mode.tsx b/app/play/[level]/test-mode.tsx
index f078c32..c0ee8b2 100644
--- a/app/play/[level]/test-mode.tsx
+++ b/app/play/[level]/test-mode.tsx
@@ -1,5 +1,6 @@
 import { useState } from "react";
 import { ReloadIcon } from "@radix-ui/react-icons";
+import { useAITesting } from "@/components/AITesting";
 import { Map } from "@/components/Map";
 import { Visualizer } from "@/components/Visualizer";
 import { Button } from "@/components/ui/button";
@@ -11,7 +12,6 @@ import {
   SelectValue,
 } from "@/components/ui/select";
 import { AI_MODELS } from "@/convex/constants";
-import { useAITesting } from "@/hooks/useAITesting";
 
 interface TestModeProps {
   level: number;
diff --git a/app/playground/page.tsx b/app/playground/page.tsx
index f2f6879..f55bd46 100644
--- a/app/playground/page.tsx
+++ b/app/playground/page.tsx
@@ -11,6 +11,7 @@ import {
 } from "lucide-react";
 import Link from "next/link";
 import { useSearchParams } from "next/navigation";
+import { useAITesting } from "@/components/AITesting";
 import { Map } from "@/components/Map";
 import { MapStatus } from "@/components/MapStatus";
 import { ModelSelector } from "@/components/ModelSelector";
@@ -29,7 +30,6 @@ import { useToast } from "@/components/ui/use-toast";
 import { api } from "@/convex/_generated/api";
 import { Id } from "@/convex/_generated/dataModel";
 import { SIGN_IN_ERROR_MESSAGE } from "@/convex/users";
-import { useAITesting } from "@/hooks/useAITesting";
 import { errorMessage } from "@/lib/utils";
 import { ZombieSurvival } from "@/simulator";
 
diff --git a/hooks/useAITesting.ts b/components/AITesting.tsx
similarity index 100%
rename from hooks/useAITesting.ts
rename to components/AITesting.tsx