X-Git-Url: http://git.veekun.com/zzz-pokedex.git/blobdiff_plain/09495b1a01a9f5a0b4f74efc9234f2afbc83277a..463601b9dddd2118d59cf3176b26843660a92e3f:/pokedex/lookup.py

diff --git a/pokedex/lookup.py b/pokedex/lookup.py
index f35e335..9e2ead6 100644
--- a/pokedex/lookup.py
+++ b/pokedex/lookup.py
@@ -58,7 +58,9 @@ class PokedexLookup(object):
         for cls in (
             tables.Ability,
             tables.Item,
+            tables.Location,
             tables.Move,
+            tables.Nature,
             tables.Pokemon,
             tables.Type,
         )
@@ -142,7 +144,7 @@ class PokedexLookup(object):
         # especially when e.g. German names are very similar to English names
         speller_entries = []
         for cls in self.indexed_tables.values():
-            q = session.query(cls)
+            q = self.session.query(cls)
 
             for row in q.yield_per(5):
                 row_key = dict(table=unicode(cls.__tablename__),