diff --git a/src/perception/base.py b/src/perception/base.py
index 4e1a3d6..c9c6307 100644
--- a/src/perception/base.py
+++ b/src/perception/base.py
@@ -29,7 +29,7 @@ from albumentations.pytorch import ToTensorV2
 OBJ_LABELS = {
     "truck": 0, "bicycle": 1, "car": 2, "motorcycle": 3, 
     "train": 4, "bus": 5, "traffic sign": 6, "rider": 7, "person": 8, 
-    "traffic light NA": 9, "traffic light R": 10, "traffic light G": 11, "traffic light B": 12
+    "traffic light NA": 9, "traffic light R": 10, "traffic light G": 11, "traffic light Y": 12
 }
 
 def use_device(GPU):