diff --git a/python/tests/beagle.py b/python/tests/beagle.py
index 99a360e28e..6c2259f02e 100644
--- a/python/tests/beagle.py
+++ b/python/tests/beagle.py
@@ -516,7 +516,7 @@ def interpolate_allele_probabilities(sm, ref_h, genotyped_pos, imputed_pos):
     rescaled_p = np.zeros_like(p)
     for i in np.arange(x):
         rescaled_p[i, :] = p[i, :] / np.sum(p[i, :])
-    return p
+    return rescaled_p
 
 
 def get_map_alleles(allele_probs):