diff --git a/jass_preprocessing/compute_score.py b/jass_preprocessing/compute_score.py
index 1a1217ac39acadc3552c68134cc57c6d86317512..fe7425c0857d34f9617bbf67b2b816881ba80e3a 100644
--- a/jass_preprocessing/compute_score.py
+++ b/jass_preprocessing/compute_score.py
@@ -25,13 +25,9 @@ def compute_z_score(mgwas):
     print(mgwas.columns)
     mgwas["computed_z"] = np.sqrt(ss.chi2.isf(mgwas['pval'], 1))
 
-    if 'z' in mgwas.columns:
-        sign_vect = np.sign(mgwas.z)
-        
-        mgwas.loc[mgwas["computed_z"].isin([np.inf]), 'computed_z'] = mgwas["z"]
 
-    elif 'beta_or_Z' in mgwas.columns:
-        sign_vect = np.sign(mgwas.beta)
+    if 'beta_or_Z' in mgwas.columns:
+        sign_vect = np.sign(mgwas.beta_or_Z)
         mgwas["computed_z"] = mgwas["computed_z"].replace(np.inf, 37.537836095576054)
     else:
         if "OR" in mgwas.columns: