diff --git a/src/routes/documents.pleading.new.tsx b/src/routes/documents.pleading.new.tsx
index 3bd20e5..12bcf6b 100644
--- a/src/routes/documents.pleading.new.tsx
+++ b/src/routes/documents.pleading.new.tsx
@@ -20,7 +20,9 @@ import { ContactPickerPopover } from "@/components/clients/contact-picker-popove
 import { supabase } from "@/integrations/supabase/client";
 import { useAuth } from "@/lib/auth";
 import { toast } from "sonner";
-import { Download, FileText, Save, Loader2, Plus, Trash2, PenLine, Users } from "lucide-react";
+import { Download, FileText, Save, Loader2, Plus, Trash2, PenLine, Users, Variable, Copy } from "lucide-react";
+import { applyVars, buildChips, buildVarMap, loadPleadingContext, type LoadedContext, type VarChip } from "@/lib/pleading-variables";
+import { Badge } from "@/components/ui/badge";
 
 export const Route = createFileRoute("/documents/pleading/new")({
   component: PleadingNewPage,