From 2f0f3ceffd4ed84cd64cbf6fef7fe355bda53dcb Mon Sep 17 00:00:00 2001
From: Shing Zhan <shing.zhan@gmail.com>
Date: Wed, 13 Sep 2023 15:23:09 +0100
Subject: [PATCH] Fix bug

---
 python/tests/beagle.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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):