diff --git a/tests/estimators/speech_recognition/test_pytorch_icefall.py b/tests/estimators/speech_recognition/test_pytorch_icefall.py
index 4806c14a08..14838a6c50 100644
--- a/tests/estimators/speech_recognition/test_pytorch_icefall.py
+++ b/tests/estimators/speech_recognition/test_pytorch_icefall.py
@@ -88,9 +88,9 @@ def test_pytorch_icefall(art_warning, expected_values, device_type):
 
         # Test transcription outputs
         hyps = []
-        hyps.append(speech_recognizer.predict(x[0]))
-        hyps.append(speech_recognizer.predict(x[1]))
-        hyps.append(speech_recognizer.predict(x[2]))
+        hyps.append(speech_recognizer.predict(xs[0]))
+        hyps.append(speech_recognizer.predict(xs[1]))
+        hyps.append(speech_recognizer.predict(xs[2]))
         print(hyps)
 
         assert (np.array(hyps) == y).all()