diff --git a/pyproject.toml b/pyproject.toml
index 9c667cf..03fdd4f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 # --- Core Metadata ---
 name = "polysolve"
-version = "0.4.0"
+version = "0.4.1"
 authors = [
   { name="Jonathan Rampersad", email="jonathan@jono-rams.work" },
 ]
diff --git a/src/polysolve/__init__.py b/src/polysolve/__init__.py
index b37b691..4bb0cd4 100644
--- a/src/polysolve/__init__.py
+++ b/src/polysolve/__init__.py
@@ -295,7 +295,8 @@ class Function:
             y_calculated = np.polyval(self.coefficients, solutions)
             error = y_calculated - y_val
             
-            ranks = np.where(error == 0, np.finfo(float).max, np.abs(1.0 / error))
+            with np.errstate(divide='ignore'):
+                ranks = np.where(error == 0, np.finfo(float).max, np.abs(1.0 / error))
             
             # Sort solutions by fitness (descending)
             sorted_indices = np.argsort(-ranks)
@@ -345,7 +346,8 @@ class Function:
         # After all generations, do one last ranking to find the best solutions
         y_calculated = np.polyval(self.coefficients, solutions)
         error = y_calculated - y_val
-        ranks = np.where(error == 0, np.finfo(float).max, np.abs(1.0 / error))
+        with np.errstate(divide='ignore'):
+            ranks = np.where(error == 0, np.finfo(float).max, np.abs(1.0 / error))
         sorted_indices = np.argsort(-ranks)
         
         # Get the top 'sample_size' solutions the user asked for