Bug fix on model counter

pkoukos · Mar 28, 2016 · 0a1d450 · 0a1d450
1 parent 9b54cd5
commit 0a1d450
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pdb_wc.py b/pdb_wc.py
@@ -77,7 +77,7 @@ def _summarize(fhandle, option):
 
     option = option
 
-    n_models = 1
+    n_models = 0
     has_hetero, has_gaps, has_double, has_insert = "No", "No", "No", "No"
     at_list, res_list, chain_list = [], [], []
     prev_resuid, prev_chainid = None, None
@@ -98,6 +98,8 @@ def _summarize(fhandle, option):
             n_models += 1
 
         if record == 'ATOM  ':
+            if not n_models:
+                n_models += 1
 
             res_uid = (line[17:20], line[21], int(line[22:26]))
             at_uid = (line[12:16], line[16], line[17:20], line[21], line[22:26])