diff --git a/python/examples/viz_atomeye.py b/python/examples/viz_atomeye.py
index d9cf4dcef274f3675d8891d6f64608893d0165a4..d7f83f274dfd57f985322d8f3c05db7d9506f779 100755
--- a/python/examples/viz_atomeye.py
+++ b/python/examples/viz_atomeye.py
@@ -42,7 +42,7 @@ lmp = lammps()
 
 lmp.file(infile)
 lmp.command("thermo %d" % nfreq)
-lmp.command("dump python all cfg %d tmp.cfg.* id type xs ys zs" % nfreq)
+lmp.command("dump python all cfg %d tmp.cfg.* mass type xs ys zs id" % nfreq)
 
 # initial 0-step run to generate dump file and image