diff --git a/Programs/Source/prf_leg.mpc b/Programs/Source/prf_leg.mpc
index 8108b965..c2cc69e9 100644
--- a/Programs/Source/prf_leg.mpc
+++ b/Programs/Source/prf_leg.mpc
@@ -27,7 +27,7 @@ class KDPreprocessing(object):
             for i in range(bit_len):
                 s, ssq = sint.get_random_square()
                 b = sint.get_random_bit()
-                t = b*(ssq - nonsq) + nonsq
+                t = b*(ssq - nonsq) + nonsq * sq
 
                 next_row = cur_row + i
                 pre_squares[next_row] = t