diff --git a/tests/dry_test/test_datasets.py b/tests/dry_test/test_datasets.py
index 01a939bc..8210b6d0 100644
--- a/tests/dry_test/test_datasets.py
+++ b/tests/dry_test/test_datasets.py
@@ -42,6 +42,7 @@
     "real_toxicity_prompts": None,
     "rte": [],
     "siqa": [],
+    "sst2": [],
     "squad": [],
     "squad_v2": [],
     "story_cloze": None,