diff --git a/examples/ngcf_example.py b/examples/ngcf_example.py
index 17846692d..b901317d5 100644
--- a/examples/ngcf_example.py
+++ b/examples/ngcf_example.py
@@ -25,8 +25,8 @@
 # Instantiate an evaluation method to split data into train and test sets.
 ratio_split = RatioSplit(
     data=data,
-    # val_size=0.1,
-    test_size=0.2,
+    val_size=0.1,
+    test_size=0.1,
     exclude_unknowns=True,
     verbose=True,
     seed=123,