diff --git a/python/pytest/test_alg.py b/python/pytest/test_alg.py
index 6155b760..18d6b701 100644
--- a/python/pytest/test_alg.py
+++ b/python/pytest/test_alg.py
@@ -134,7 +134,8 @@ def test_binomial():
         test_data = load_data("binomial_test")
 
         def assert_reg(coef, fit_intercept=True, rel=0.01, abs=0.01):
-            if sys.version_info[0] + 0.1 * sys.version_info[1] < 3.6:
+            sys_version_num = sys.version_info[0] + 0.1 * sys.version_info[1]
+            if sys_version_num < 3.6 or sys_version_num >= 3.9:
                 return
             nonzero = np.nonzero(coef)[0]
             new_x = data.x[:, nonzero]