diff --git a/src/ctx_to_lora/data_utils.py b/src/ctx_to_lora/data_utils.py
index c7387f2..b25b377 100644
--- a/src/ctx_to_lora/data_utils.py
+++ b/src/ctx_to_lora/data_utils.py
@@ -36,6 +36,7 @@ DS_KWARGS = {
     "squad": dict(
         train=dict(path="rajpurkar/squad", split="train[900:]"),
         validation=dict(path="rajpurkar/squad", split="train[:900]"),
+        test=dict(path="rajpurkar/squad", split="validation"),
     ),
     "fw_qa_tiny": dict(
         train=dict(