devision by zero

pdf_generator/tex_generation/template_engine.py

@@ -119,6 +119,8 @@ def _count_mitigations(summary):
         count = _count_this_mitigation(summary, mitigation)
         if count != 0:
             return count
+    if count == 0:
+        count = 1
     return count
 
 
@@ -127,8 +129,6 @@ def _count_this_mitigation(summary, mitigation):
     for selected_summary in summary:
         if mitigation in selected_summary:
             count += len(summary[selected_summary])
-    if count == 0:
-        count = 1
     return count