diff --git a/configs/nvm/sweep.py b/configs/nvm/sweep.py
index d5d23ad76a0..ab77768e08e 100644
--- a/configs/nvm/sweep.py
+++ b/configs/nvm/sweep.py
@@ -59,7 +59,7 @@
 
 # Use a single-channel DDR3-1600 x64 (8x8 topology) by default
 parser.add_argument(
-    "--nvm-type",
+    "--mem-type",
     default="NVM_2400_1x64",
     choices=ObjectList.mem_list.get_names(),
     help="type of memory to use",
@@ -212,7 +212,7 @@ def trace():
                 nbr_banks,
                 bank,
                 addr_map,
-                args.dram_ranks,
+                args.nvm_ranks,
             )
     yield system.tgen.createExit(0)
 
diff --git a/configs/nvm/sweep_hybrid.py b/configs/nvm/sweep_hybrid.py
index 669f847eb1b..82a4a6124e2 100644
--- a/configs/nvm/sweep_hybrid.py
+++ b/configs/nvm/sweep_hybrid.py
@@ -143,7 +143,7 @@
 
 # the following assumes that we are using the native controller
 # with NVM and DRAM interfaces, check to be sure
-if not isinstance(system.mem_ctrls[0], m5.objects.HeteroMemCtrl):
+if not isinstance(system.mem_ctrls[0], m5.objects.MemCtrl):
     fatal("This script assumes the controller is a HeteroMemCtrl subclass")
 if not isinstance(system.mem_ctrls[0].dram, m5.objects.DRAMInterface):
     fatal("This script assumes the first memory is a DRAMInterface subclass")