diff --git a/tests/test_prompts.py b/tests/test_prompts.py
index 9296a03..115ef43 100644
--- a/tests/test_prompts.py
+++ b/tests/test_prompts.py
@@ -126,7 +126,7 @@ def preprocess_qa(example):
         self.question_answering = load_dataset("squad", split="train").flatten()\
             .rename_column("answers.text", "answer").map(preprocess_qa).select([1, 2, 3])
         self.ner = load_dataset("conll2003", split="train").select([1, 2, 3])
-        self.translation = load_dataset("opus100", language_pair="de-nl", split="test").flatten()\
+        self.translation = load_dataset("opus100", "de-nl", split="test").flatten()\
             .rename_columns({"translation.de": "german", "translation.nl": "dutch"}).select([1, 2, 3])
 
     def test_translation(self):