diff --git a/hydra_plugins/hypersweeper/hypersweeper_sweeper.py b/hydra_plugins/hypersweeper/hypersweeper_sweeper.py
index 7f00888..dd35202 100644
--- a/hydra_plugins/hypersweeper/hypersweeper_sweeper.py
+++ b/hydra_plugins/hypersweeper/hypersweeper_sweeper.py
@@ -350,7 +350,10 @@ def record_iteration(self, performances, configs, budgets):
         for i in range(len(configs)):
             self.history["configs"].append(configs[i])
             self.history["performances"].append(performances[i])
-            self.history["budgets"].append(budgets[i])
+            if budgets[i] is not None:
+                self.history["budgets"].append(budgets[i])
+            else:
+                self.history["budgets"].append(self.max_budget)
         self.iteration += 1
 
         if self.wandb_project: