diff --git a/samples/python2/digits.py b/samples/python2/digits.py
index b1688a8a07aa51fdf8db5fc72a19468e7bdc02c9..04397a14317af42655785ad2819ad5dac5837ad7 100755
--- a/samples/python2/digits.py
+++ b/samples/python2/digits.py
@@ -99,7 +99,7 @@ class SVM(StatModel):
         self.model.train(samples, cv2.ml.ROW_SAMPLE, responses)
 
     def predict(self, samples):
-        return self.model.predict(samples)[1][0].ravel()
+        return self.model.predict(samples)[1].ravel()
 
 
 def evaluate_model(model, digits, samples, labels):