Various tweaks and fixes

1 files changed, 3 insertions, 2 deletions

diff --git a/gitlog.py b/gitlog.py
index 90975ef..8b8d523 100644
--- a/gitlog.py
+++ b/gitlog.py
@@ -196,8 +196,9 @@ def grabpatch(input):
     while state != S_DONE:
         line = getline(input)
         if line is None:
-            if state != S_TAGS:
-                print 'Ran out of patch'
+            if state != S_NUMSTAT:
+                print 'Ran out of patch', state
+                return None
             return p
         state = grabbers[state](p, line, input)
     return p