From 4fc462c4d9c4cf750f4552629d8b9d8c1334f3e9 Mon Sep 17 00:00:00 2001
From: Miroslav Stampar <miroslav.stampar@gmail.com>
Date: Wed, 18 Jul 2012 14:09:04 +0200
Subject: [PATCH] Minor update for an Issue #105

---
 lib/core/wordlist.py | 11 +++--------
 1 file changed, 3 insertions(+), 8 deletions(-)

diff --git a/lib/core/wordlist.py b/lib/core/wordlist.py
index 488488515..482677f4a 100644
--- a/lib/core/wordlist.py
+++ b/lib/core/wordlist.py
@@ -29,10 +29,7 @@ class Wordlist:
         if self.index > len(self.filenames):
             raise StopIteration
         elif self.index == len(self.filenames):
-            if not self.proc_id:
-                self.iter = iter(self.custom)
-            else:
-                raise StopIteration
+            self.iter = iter(self.custom)
         else:
             current = self.filenames[self.index]
             self.fp = open(current, "r")
@@ -48,15 +45,13 @@ class Wordlist:
     def next(self):
         retVal = None
         while True:
+            self.counter += 1
             try:
                 retVal = self.iter.next().rstrip()
             except StopIteration:
                 self.adjust()
                 retVal = self.iter.next().rstrip()
-            if not self.proc_count:
-                break
-            self.counter += 1
-            if self.counter % self.proc_count == self.proc_id:
+            if not self.proc_count or self.counter % self.proc_count == self.proc_id:
                 break
         return retVal