diff --git a/experiment/scheduler.py b/experiment/scheduler.py
index 2bd999874..b189eabb6 100644
--- a/experiment/scheduler.py
+++ b/experiment/scheduler.py
@@ -791,7 +791,7 @@ def render_startup_script_template(  # pylint: disable=too-many-arguments
 def create_trial_instance(  # pylint: disable=too-many-arguments
         fuzzer: str,
         benchmark: str,
-        trial_id: int,       
+        trial_id: int,
         experiment_config: dict,
         preemptible: bool,
         cpuset=None,
diff --git a/experiment/test_scheduler.py b/experiment/test_scheduler.py
index c486dd183..4832ab20f 100644
--- a/experiment/test_scheduler.py
+++ b/experiment/test_scheduler.py
@@ -114,7 +114,7 @@ def test_create_trial_instance(benchmark, expected_image, expected_target,
 -e BENCHMARK={benchmark} \\
 -e EXPERIMENT=test-experiment \\
 -e TRIAL_ID=9 \\
--e TRIAL_GROUP_NUM=0 \\
+-e MAX_TOTAL_TIME=86400 \\
 -e TRIAL_GROUP_NUM=0 \\
 -e MICRO_EXPERIMENT=False \\
 -e SNAPSHOT_PERIOD=900 \\