Performance

1 files changed, 1 insertions, 1 deletions

diff --git a/pypride.py b/pypride.py
index b30b565..c011f74 100644
--- a/pypride.py
+++ b/pypride.py
@@ -239,7 +239,7 @@ def matrixMultiply(matrix, input):
             input as Int
     Output: Int"""
 
-    mult = [ mapXor([c * ((input >> (15 - c_i)) & 0x1) for c_i, c in reversed(list(enumerate(r)))]) for r in matrix ]
+    mult = [ mapXor([1 for c_i, c in enumerate(r) if c == 1 and input & (0x1 << (15 - c_i))]) for r in matrix ]
     return sum([(1 << (15-i)) * v for i,v in enumerate(mult)])
 
 def roundKey(key, i):