1 files changed, 3 insertions, 8 deletions

diff --git a/compact_pm_ovd.py b/compact_pm_ovd.py
index 4670de0..10151d4 100755
--- a/compact_pm_ovd.py
+++ b/compact_pm_ovd.py
@@ -57,14 +57,9 @@ class WSSCompactor():
             wstr += 'wss=' + str(wss) + ':(max, avg), '
         csvf.write(wstr[0:-2] + '\n')
 
-        tkeys = self.ovhead_table.keys()
-        for i in range(0,len(tkeys)):
-            tkeys[i] = int(tkeys[i])
-        tkeys.sort()
-
-        for key in tkeys:
-            wstr = str(key) + ', '
-            for value in self.ovhead_table[str(key)]:
+        for key in sorted(self.ovhead_table.keys(), key=int):
+            wstr = key + ', '
+            for value in self.ovhead_table[key]:
                 if float(value) > 0:
                     wstr += value + ', '
                 else: