diff --git a/kernel_two_sample_test.py b/kernel_two_sample_test.py
index ddf950a..23fe49c 100644
--- a/kernel_two_sample_test.py
+++ b/kernel_two_sample_test.py
@@ -31,7 +31,7 @@ def compute_null_distribution(K, m, n, iterations=10000, verbose=False,
             stdout.flush()
         idx = rng.permutation(m+n)
         K_i = K[idx, idx[:, None]]
-        mmd2u_null[i] = MMD2u(K_i, m, n)
+        mmd2u_null[i] = (n+m) * MMD2u(K_i, m, n)
 
     if verbose:
         print("")
@@ -53,7 +53,7 @@ def compute_null_distribution_given_permutations(K, m, n, permutation,
     for i in range(iterations):
         idx = permutation[i]
         K_i = K[idx, idx[:, None]]
-        mmd2u_null[i] = MMD2u(K_i, m, n)
+        mmd2u_null[i] = (n+m) * MMD2u(K_i, m, n)
 
     return mmd2u_null