diff --git a/AppImage/components/network-traffic-chart.tsx b/AppImage/components/network-traffic-chart.tsx
index 8515db8..1e47fb3 100644
--- a/AppImage/components/network-traffic-chart.tsx
+++ b/AppImage/components/network-traffic-chart.tsx
@@ -3,8 +3,8 @@
 import { useState, useEffect } from "react"
 import { AreaChart, Area, XAxis, YAxis, CartesianGrid, Tooltip, ResponsiveContainer, Legend } from "recharts"
 import { Loader2 } from 'lucide-react'
-import { fetchApi } from "@/lib/api-config"
-import { getNetworkUnit } from "@/lib/format-network"
+import { fetchApi } from "../lib/api-config"
+import { getNetworkUnit } from "../lib/format-network"
 
 interface NetworkMetricsData {
   time: string
diff --git a/AppImage/components/virtual-machines.tsx b/AppImage/components/virtual-machines.tsx
index 593b9e7..7e52a14 100644
--- a/AppImage/components/virtual-machines.tsx
+++ b/AppImage/components/virtual-machines.tsx
@@ -11,8 +11,8 @@ import { Dialog, DialogContent, DialogHeader, DialogTitle } from "./ui/dialog"
 import { Server, Play, Square, Cpu, MemoryStick, HardDrive, Network, Power, RotateCcw, StopCircle, Container, ChevronDown, ChevronUp } from 'lucide-react'
 import useSWR from "swr"
 import { MetricsView } from "./metrics-dialog"
-import { formatStorage } from "@/lib/utils" // Import formatStorage utility
-import { formatNetworkTraffic, getNetworkUnit } from "@/lib/format-network"
+import { formatStorage } from "../lib/utils"
+import { formatNetworkTraffic, getNetworkUnit } from "../lib/format-network"
 import { fetchApi } from "../lib/api-config"
 
 interface VMData {
@@ -1238,7 +1238,8 @@ export function VirtualMachines() {
                                       </>
                                     ) : (
                                       <>
-                                        <ChevronDown className="h-3 w-3 mr-1" />+ Info
+                                        <ChevronDown className="h-3 w-3 mr-1" />
+                                        + Info
                                       </>
                                     )}
                                   </Button>
diff --git a/AppImage/lib/format-network.ts b/AppImage/lib/format-network.ts
index 39e645b..43cca41 100644
--- a/AppImage/lib/format-network.ts
+++ b/AppImage/lib/format-network.ts
@@ -42,7 +42,7 @@ export function formatNetworkTraffic(
 export function getNetworkUnit(): NetworkUnit {
   if (typeof window === 'undefined') return 'Bytes';
   
-  const stored = localStorage.getItem('networkUnit');
+  const stored = localStorage.getItem('proxmenux-network-unit');
   return stored === 'Bits' ? 'Bits' : 'Bytes';
 }