Fixed potential bug in the count function.

1 files changed, 2 insertions, 1 deletions

diff --git a/sta.py b/sta.py
index a73de40..684d0ac 100755
--- a/sta.py
+++ b/sta.py
@@ -86,10 +86,11 @@ class Trace:
 
     def print_count(self,verbose=False):
         """Prints the number of records in the trace."""
+        self.iter = list(self.iter)
         print("Total Count: " + str(len(self.iter)))
         if verbose:
             counts = {1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0}
-            for record in self.iter:
+            for record in list(self.iter):
                 type = record['type']
                 counts[type] += 1
             for key in counts.keys():