X-Git-Url: http://git.veekun.com/zzz-pokedex.git/blobdiff_plain/958a4f9f89bbc405ff641a157fe3fb465e737f07..d2b4c6a2892886fc9ad0998998bb1c65ba618e45:/pokedex/savefile.py

diff --git a/pokedex/savefile.py b/pokedex/savefile.py
index f64eec4..a3c4090 100644
--- a/pokedex/savefile.py
+++ b/pokedex/savefile.py
@@ -8,7 +8,8 @@ Kudos to LordLandon for his pkmlib.py, from which this module was originally
 derived.
 """
 
-import itertools, struct
+import struct
+from pokedex.util import permutations
 
 def pokemon_prng(seed):
     u"""Creates a generator that simulates the main Pokémon PRNG."""
@@ -79,7 +80,7 @@ class PokemonSave(object):
 
     ### Utility methods
 
-    shuffle_orders = list( itertools.permutations(range(4)) )
+    shuffle_orders = list( permutations(range(4)) )
 
     @classmethod
     def shuffle_chunks(cls, words, reverse=False):