diff --git a/ANN_Training_Data_Generator.py b/ANN_Training_Data_Generator.py
index 8f6114eef952edb4d8d3186f8abc92393b7973eb..f81cce1804f31dc0b9ef94efe965e04daefd4e8a 100644
--- a/ANN_Training_Data_Generator.py
+++ b/ANN_Training_Data_Generator.py
@@ -3,6 +3,7 @@
 @author: Soraya Terrab (sorayaterrab), Laura C. Kühle
 
 TODO: Improve '_generate_cell_data'
+TODO: Change set name from 'validation' to 'valid' -> Done
 
 """
 
@@ -198,7 +199,7 @@ functions = [{'function': Initial_Condition.Sine(boundary[0], boundary[1], {}),
              {'function': Initial_Condition.HeavisideTwoSided(boundary[0], boundary[1], {}), 'config': {
                  'adjustment': 0}}]
 
-generator = TrainingDataGenerator(functions, distribution={'train': 0.727, 'validation': 0.243, 'test': 0.03},
+generator = TrainingDataGenerator(functions, distribution={'train': 0.727, 'valid': 0.243, 'test': 0.03},
                                   left_bound=boundary[0], right_bound=boundary[1])
 # generator = TrainingDataGenerator(functions, left_bound=boundary[0], right_bound=boundary[1])