diff --git a/samples/python2/digits.py b/samples/python2/digits.py
index b1688a8a07..04397a1431 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):