diff --git a/binary_classification_workflow.py b/binary_classification_workflow.py
index 8eb755c0c9b3d50ee3cbce64a405e3b127505e47..1150b3739a343e65076f0e24106c5a4ec1d0decb 100644
--- a/binary_classification_workflow.py
+++ b/binary_classification_workflow.py
@@ -61,7 +61,7 @@ def prepare_training_data(data, target, use_pca=True, threshold_variance_ratio=0
 
     if use_pca:
 
-        df, explainable_ratios= feature_selection(data, target, threshold_variance_ratio=0.90)
+        df, explainable_ratios= feature_selection(data, target, threshold_variance_ratio)
         #Explainable plots
         fig, axes = plt.subplots(1, 2, figsize=(10, 4)) 
         axes[0].scatter(df.loc[df['classification'] == 0, 'PCA1'], df.loc[df['classification'] == 0, 'PCA2'], color='red', label="CKD")