diff --git a/GPRL/algorithms.py b/GPRL/algorithms.py
index 020a6521e647dc8be55f135d4957221035e62cf1..6f110dad9519ed5eea21b9da4a2235dfdbba6a12 100644
--- a/GPRL/algorithms.py
+++ b/GPRL/algorithms.py
@@ -54,7 +54,7 @@ def eaMuPlusLambdaUCB(population, toolbox, simulation_budget, parallel_update, m
     # Begin the generational process
     for gen in range(1, ngen + 1):
         if budget_scheduler is not None:
-            simulation_budget, parallel_update = budget_scheduler(ngen, population, simulation_budget, parallel_update)
+            simulation_budget, parallel_update = budget_scheduler(gen, population, simulation_budget, parallel_update)
         # Vary the population
         offspring = algorithms.varOr(population, toolbox, lambda_, cxpb, mutpb)