diff --git a/tests/test_supervised.py b/tests/test_supervised.py
index e1d9098d2..b8bec4f45 100644
--- a/tests/test_supervised.py
+++ b/tests/test_supervised.py
@@ -18,7 +18,7 @@ def test_supervised_eval():
     if tokenizer.pad_token is None:
         tokenizer.pad_token = tokenizer.eos_token
 
-    output = preprocess_supervised_example(examples, tokenizer)
+    output = preprocess_supervised_example(examples, tokenizer, "input", "output")
     assert len(output["input_ids"][0]) == output["sources_len"][0] + 1
 
     ex = {