diff --git a/python/lammps.py b/python/lammps.py
index d428a097a81589b3bae8c18355a9bf40fd2db1ca..a512efdcda71abfed9209aa44978ad3040c9f0b0 100644
--- a/python/lammps.py
+++ b/python/lammps.py
@@ -556,9 +556,10 @@ def get_thermo_data(output):
     runs = []
     columns = []
     in_run = False
+    current_run = {}
 
     for line in lines:
-        if line.startswith("Memory usage per processor"):
+        if line.startswith("Per MPI rank memory allocation"):
             in_run = True
         elif in_run and len(columns) == 0:
             # first line after memory usage are column names