diff --git a/scripts/profile-memory.py b/scripts/profile-memory.py
index 55f2e76a..349f6ee9 100644
--- a/scripts/profile-memory.py
+++ b/scripts/profile-memory.py
@@ -60,7 +60,8 @@ def main():
 
         process = psutil.Process(os.getpid())
         print("  duration: %0.02fs" % (t1 - t0))
-        print("  rss: %.1f KiB" % (process.memory_info().rss / 1024))
+        print("  rss: %.1f MiB" % (process.memory_info().rss / 1024 / 1024))
+        print("  vms: %.1f MiB" % (process.memory_info().vms / 1024 / 1024))
 
     print("done.")
     gc.collect()