diff --git a/test/unit/model/regressors/test_flatliner.py b/test/unit/model/regressors/test_flatliner.py
index 3fcf5a5ad..32fe33a6a 100644
--- a/test/unit/model/regressors/test_flatliner.py
+++ b/test/unit/model/regressors/test_flatliner.py
@@ -49,7 +49,7 @@ def test_quantile_fit(self):
 
         result: np.ndarray = model.predict(train_input.iloc[:, 1:])
 
-        self.assertEquals(len(result), len(train_input.iloc[:, 1:]))
+        self.assertEqual(len(result), len(train_input.iloc[:, 1:]))
         self.assertTrue((result == 0).all())
 
     def test_get_feature_names_from_linear(self):