diff --git a/tests/assets/widerface_dataset/dataset_meta.json b/tests/assets/widerface_dataset/dataset_meta.json
index 0983e2f489..55b4b8b4c1 100644
--- a/tests/assets/widerface_dataset/dataset_meta.json
+++ b/tests/assets/widerface_dataset/dataset_meta.json
@@ -1,3 +1,3 @@
 {
-"label_map": {"0": "Parade", "1": "Handshaking"}
+"labels": ["face"]
 }
diff --git a/tests/test_widerface_format.py b/tests/test_widerface_format.py
index 974d72c370..926985579f 100644
--- a/tests/test_widerface_format.py
+++ b/tests/test_widerface_format.py
@@ -430,8 +430,8 @@ def test_can_import(self):
                                 "pose": "0",
                                 "invalid": "0",
                             },
+                            label=0,
                         ),
-                        Label(0),
                     ],
                 ),
                 DatasetItem(
@@ -452,6 +452,7 @@ def test_can_import(self):
                                 "pose": "0",
                                 "invalid": "0",
                             },
+                            label=0,
                         ),
                         Bbox(
                             5,
@@ -466,8 +467,8 @@ def test_can_import(self):
                                 "pose": "0",
                                 "invalid": "0",
                             },
+                            label=0,
                         ),
-                        Label(1),
                     ],
                 ),
                 DatasetItem(
@@ -488,6 +489,7 @@ def test_can_import(self):
                                 "pose": "2",
                                 "invalid": "0",
                             },
+                            label=0,
                         ),
                         Bbox(
                             3,
@@ -502,6 +504,7 @@ def test_can_import(self):
                                 "pose": "0",
                                 "invalid": "0",
                             },
+                            label=0,
                         ),
                         Bbox(
                             5,
@@ -516,12 +519,12 @@ def test_can_import(self):
                                 "pose": "2",
                                 "invalid": "0",
                             },
+                            label=0,
                         ),
-                        Label(0),
                     ],
                 ),
             ],
-            categories=["Parade", "Handshaking"],
+            categories=["face"],
         )
 
         dataset = Dataset.import_from(DUMMY_DATASET_DIR, "wider_face")