diff --git a/grains/init.sls b/grains/init.sls
index ebb3c1f..b0b40f7 100644
--- a/grains/init.sls
+++ b/grains/init.sls
@@ -40,7 +40,7 @@ class GrainMaker:
 
   def _parseGrainValues(self, grain):
     # Parses individual grains and their values passed from the pillar
-    grainKey = grain.keys()[0]
+    grainKey = next(iter(grain.keys()))
     if grain[grainKey]:
       for value in grain[grainKey]:
         self._addToGrains(grain, grainKey, value)