diff --git a/maskrcnn_benchmark/data/datasets/bdd100k.py b/maskrcnn_benchmark/data/datasets/bdd100k.py
index 89d18da5e..2093dc48c 100644
--- a/maskrcnn_benchmark/data/datasets/bdd100k.py
+++ b/maskrcnn_benchmark/data/datasets/bdd100k.py
@@ -81,7 +81,7 @@ def __getitem__(self, idx):
         classes = []
         for label in annotations['labels']:
             # TODO: further filter annotations if needed
-
+            classes += [label['category']]
             boxes += [
                 label['box2d']['x1'],
                 label['box2d']['y1'],