diff --git a/tools/python/pizza/dump.py b/tools/python/pizza/dump.py
index 3dabc9b1b4..d57bdf63a3 100644
--- a/tools/python/pizza/dump.py
+++ b/tools/python/pizza/dump.py
@@ -657,11 +657,14 @@ class dump:
 
       atoms = snap.atoms
       nvalues = len(atoms[0])
+      keys = dict()
+      for pair in self.names.items():
+        keys[pair[1]] = pair[0]
       for i in range(snap.natoms):
         if not snap.aselect[i]: continue
         line = ""
         for j in range(nvalues):
-          if (j < 2):
+          if keys[j] == 'id' or keys[j] == 'type' or keys[j] == 'mol':
             line += str(int(atoms[i][j])) + " "
           else:
             line += str(atoms[i][j]) + " "