diff --git a/bob/bio/face/test/test_baselines.py b/bob/bio/face/test/test_baselines.py
index ab1892d81181931c54414e57c95107a95bdbded2..4bff3b8ed6880986f49d6ad79b7046a42c06906c 100644
--- a/bob/bio/face/test/test_baselines.py
+++ b/bob/bio/face/test/test_baselines.py
@@ -42,7 +42,7 @@ def get_fake_sample_set(face_size=(160, 160), purpose="bioref"):
                 )
             ],
             key=key,
-            subject=key,
+            reference_id=key,
             references=["1"],
         )
     ]
@@ -54,7 +54,7 @@ def get_fake_samples_for_training():
     annotations = {"reye": (131, 176), "leye": (222, 170)}
 
     return [
-        Sample(x, key=str(i), subject=str(i), annotations=annotations)
+        Sample(x, key=str(i), reference_id=str(i), annotations=annotations)
         for i, x in enumerate(data)
     ]