diff --git a/src/components/battery/BatteryCard.tsx b/src/components/battery/BatteryCard.tsx
index af3c2f2..effcc8d 100644
--- a/src/components/battery/BatteryCard.tsx
+++ b/src/components/battery/BatteryCard.tsx
@@ -9,7 +9,8 @@ import { Input } from '@/components/ui/Input';
import { Select } from '@/components/ui/Select';
import { useToast } from '@/components/ui/Toast';
import { useRouter } from 'next/navigation';
-import { Battery, BatteryCharging, Monitor, Trash2, ArrowRight } from 'lucide-react';
+import { Battery, BatteryCharging, Monitor, Trash2, ArrowRight, Pencil } from 'lucide-react';
+import { EditBatteryModal } from './EditBatteryModal';
interface Device {
id: number;
@@ -36,6 +37,7 @@ export function BatteryCard({ battery, devices }: BatteryCardProps) {
const router = useRouter();
const { showToast } = useToast();
const [actionModal, setActionModal] = useState<'charge' | 'available' | 'assign' | 'delete' | null>(null);
+ const [showEditModal, setShowEditModal] = useState(false);
const [count, setCount] = useState('1');
const [deviceId, setDeviceId] = useState('');
const [loading, setLoading] = useState(false);
@@ -148,6 +150,14 @@ export function BatteryCard({ battery, devices }: BatteryCardProps) {
Total: {total} batteries
+