diff --git a/engine.py b/engine.py
index af12677f..bd425e3c 100644
--- a/engine.py
+++ b/engine.py
@@ -2,7 +2,7 @@
 
 import os
 
-os.environ["FLAGS_USE_SYSTEM_ALLOCATOR"] = "1"
+os.environ["FLAGS_use_system_allocator"] = "1"
 os.environ["NVIDIA_TF32_OVERRIDE"] = "0"
 
 import argparse
diff --git a/engineV2.py b/engineV2.py
index ffe3a605..cc49b47b 100644
--- a/engineV2.py
+++ b/engineV2.py
@@ -39,7 +39,7 @@
 
 from tester.api_config.log_writer import *
 
-os.environ["FLAGS_USE_SYSTEM_ALLOCATOR"] = "1"
+os.environ["FLAGS_use_system_allocator"] = "1"
 os.environ["NVIDIA_TF32_OVERRIDE"] = "0"
 
 VALID_TEST_ARGS = {