diff --git a/torchprof/display.py b/torchprof/display.py
index 5c3aaf5..7d97706 100644
--- a/torchprof/display.py
+++ b/torchprof/display.py
@@ -9,10 +9,10 @@
         "cpu_total",
         "self_cuda_total",
         "cuda_total",
-        "cpu_memory",
         "self_cpu_memory",
-        "cuda_memory",
+        "cpu_memory",
         "self_cuda_memory",
+        "cuda_memory",
         "occurrences",
     ],
 )