diff --git a/src/gentropy/susie_finemapper.py b/src/gentropy/susie_finemapper.py
index 8d25f533e..03e1c2a0b 100644
--- a/src/gentropy/susie_finemapper.py
+++ b/src/gentropy/susie_finemapper.py
@@ -75,6 +75,7 @@ def susie_finemapper_one_studylocus_row(
             GWAS.df.withColumn("z", f.col("beta") / f.col("standardError"))
             .withColumn("chromosome", f.split(f.col("variantId"), "_")[0])
             .withColumn("position", f.split(f.col("variantId"), "_")[1])
+            .filter(f.col("studyId") == studyId)
             .filter(f.col("z").isNotNull())
         )
 
diff --git a/tests/gentropy/method/test_susie_inf.py b/tests/gentropy/method/test_susie_inf.py
index e7a5d219f..0129a5934 100644
--- a/tests/gentropy/method/test_susie_inf.py
+++ b/tests/gentropy/method/test_susie_inf.py
@@ -72,8 +72,8 @@ def test_SUSIE_inf_convert_to_study_locus(
         L1 = SusieFineMapperStep.susie_inf_to_studylocus(
             susie_output=susie_output,
             session=session,
-            _studyId="sample_id",
-            _region="sample_region",
+            studyId="sample_id",
+            region="sample_region",
             variant_index=gwas_df,
             cs_lbf_thr=2,
         )