diff --git a/examples/plot_01_survival_analysis.py b/examples/plot_01_survival_analysis.py
index 969eb75..316bbfd 100644
--- a/examples/plot_01_survival_analysis.py
+++ b/examples/plot_01_survival_analysis.py
@@ -48,8 +48,7 @@
 # %%
 from sklearn.model_selection import train_test_split
 
-X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
-X_train, X_val, y_train, y_val = train_test_split(X_train, y_train, test_size=0.2)
+X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3)
 
 # %%
 #