diff --git a/samples/python/digits.py b/samples/python/digits.py
index 5cc98b713c..16d76a13b0 100755
--- a/samples/python/digits.py
+++ b/samples/python/digits.py
@@ -109,7 +109,7 @@ def evaluate_model(model, digits, samples, labels):
 
     confusion = np.zeros((10, 10), np.int32)
     for i, j in zip(labels, resp):
-        confusion[i, j] += 1
+        confusion[i, int(j)] += 1
     print('confusion matrix:')
     print(confusion)
     print()