diff --git a/scripts/sweep_configs/sweep_config_example.json b/scripts/sweep_configs/sweep_config_example.json
index ba801da6..394280d6 100644
--- a/scripts/sweep_configs/sweep_config_example.json
+++ b/scripts/sweep_configs/sweep_config_example.json
@@ -1,7 +1,7 @@
 {
     "method": "random",
     "metric": {
-        "name": "Train/mean_reward",
+        "name": "rewards/total_rewards",
         "goal": "maximize"
     },
     "run_cap": 10,
diff --git a/scripts/sweep_configs/sweep_config_test_sweeps.json b/scripts/sweep_configs/sweep_config_test_sweeps.json
index 7ddb6360..41441bab 100644
--- a/scripts/sweep_configs/sweep_config_test_sweeps.json
+++ b/scripts/sweep_configs/sweep_config_test_sweeps.json
@@ -1,7 +1,7 @@
 {
     "method": "random",
     "metric": {
-        "name": "Train/mean_reward",
+        "name": "rewards/total_rewards",
         "goal": "maximize"
     },
     "run_cap": 10,