diff --git a/setup.py b/setup.py
index d371ed5bd..907f1fe50 100644
--- a/setup.py
+++ b/setup.py
@@ -66,8 +66,7 @@ setuptools.setup(
         "catboost": ["catboost>=0.26"],
         "blendsearch": ["optuna==2.8.0"],
         "ray": [
-            "ray[tune]==1.6.0",
-            "pyyaml<5.3.1",
+            "ray[tune]~=1.10",
         ],
         "azureml": [
             "azureml-mlflow",
diff --git a/test/nlp/test_autohf_regression.py b/test/nlp/test_autohf_regression.py
index 1854f7214..762091b01 100644
--- a/test/nlp/test_autohf_regression.py
+++ b/test/nlp/test_autohf_regression.py
@@ -76,7 +76,7 @@ def test_regression():
     }
 
     ray.shutdown()
-
+    ray.init()
     automl.fit(
         X_train=X_train, y_train=y_train, X_val=X_val, y_val=y_val, **automl_settings
     )