1 files changed, 7 insertions, 1 deletions
diff --git a/util.py b/util.py
index a1575db7..2ef986ef 100644
--- a/util.py
+++ b/util.py
@@ -19,6 +19,8 @@ def unfussy_reader(reader):
   while True:
     try:
       yield next(reader)
+    except StopIteration:
+      return
     except csv.Error:
       print(csv.Error)
       # log the problem or whatever
@@ -109,7 +111,11 @@ class DbPaper(object):
 class RawPaper(object):
   def __init__(self, paper_id):
     self.paper_id = paper_id
-    self.data = read_json(paper_path('raw_papers', paper_id))['paper']
+    data = read_json(paper_path('raw_papers', paper_id))
+    if 'paper' not in data:
+      print(data)
+      return None
+    self.data = data['paper']
   @property
   def title(self):
     return self.data['title']['text']