diff --git a/bob/pad/face/test/test_transformers.py b/bob/pad/face/test/test_transformers.py
index 90d3209cf79f205a334f19b6bb21f93b4988492c..082301ffede60e0f08a0f948a383dba352c2b8e8 100644
--- a/bob/pad/face/test/test_transformers.py
+++ b/bob/pad/face/test/test_transformers.py
@@ -14,6 +14,6 @@ def test_video_to_frames():
     samples = [mario.Sample(v, key=i) for i, v in enumerate(video_container)]
     frame_samples = VideoToFrames().transform(samples)
     assert len(frame_samples) == 4
-    assert all(s.key == 0 for s in frame_samples)
+    assert all("0_" in s.key for s in frame_samples)
     assert [s.data for s in frame_samples] == [0, 1, 2, 3]
     assert [s.frame_id for s in frame_samples] == [0, 1, 2, 4]
diff --git a/requirements.txt b/requirements.txt
index 3bd51dbb637617381a2d369faf6d780df623d362..85fa5c8bb5784c65bc7e99dfc4d70780420f6a46 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,4 +9,4 @@ bob.bio.video
 scikit-learn
 scikit-image
 imageio-ffmpeg
-pytorch
+torch