diff --git a/main.py b/main.py
index 0b161e2..ac079f3 100644
--- a/main.py
+++ b/main.py
@@ -52,13 +52,11 @@ def diskPollWorker(stopEvent: Event, eventQueue: Queue) -> None:
             return
 
 
-def ledWorker(stopEvent: Event, state) -> None:
+def ledWorker(stopEvent: Event, diskState: DiskState) -> None:
     yellowCtr = 0
     previousState: DiskState | None = None
 
     while not stopEvent.is_set():
-        diskState: DiskState = state.diskState
-
         if diskState != previousState:
             yellowCtr = 0
             previousState = diskState