diff --git a/src/gentropy/l2g.py b/src/gentropy/l2g.py
index cc1b5a5d1..b585166c1 100644
--- a/src/gentropy/l2g.py
+++ b/src/gentropy/l2g.py
@@ -84,7 +84,9 @@ def __init__(
             gene_index=gene_index,
         )
 
-        fm = credible_set.build_feature_matrix(features_list, features_input_loader)
+        fm = credible_set.filter(f.col("studyType") == "gwas").build_feature_matrix(
+            features_list, features_input_loader
+        )
         fm._df.write.mode(session.write_mode).parquet(feature_matrix_path)