diff --git a/common/compute.py b/common/compute.py
index 38e7bd5..48f0479 100644
--- a/common/compute.py
+++ b/common/compute.py
@@ -1,9 +1,11 @@
 import numpy as np
+
+
 def division(a, b, n=2):
     res = 0
     try:
         res = round(a / b, n)
-        if np.isnan(res):
+        if np.isnan(res) or res == np.inf:
             res = 0
     except ZeroDivisionError:
         pass