diff --git a/src/components/cases/time-tab.tsx b/src/components/cases/time-tab.tsx
index 5623590..0c3b79b 100644
--- a/src/components/cases/time-tab.tsx
+++ b/src/components/cases/time-tab.tsx
@@ -7,8 +7,9 @@ import { Textarea } from "@/components/ui/textarea";
 import { Checkbox } from "@/components/ui/checkbox";
 import { supabase } from "@/integrations/supabase/client";
 import { useAuth } from "@/lib/auth";
-import { Plus, Trash2, Loader2, FilePlus, PlusCircle } from "lucide-react";
+import { Plus, Trash2, Loader2, FilePlus, PlusCircle, CheckCircle2, Undo2 } from "lucide-react";
 import { formatCurrency, formatDate } from "@/lib/format";
+import { ensureMarkedInvoicedPlaceholder } from "@/lib/invoice-generation";
 import { roundToSixth } from "@/lib/timer";
 import { toast } from "sonner";
 import {