diff --git a/mlos_bench/mlos_bench/tests/optimizers/toy_optimization_loop_test.py b/mlos_bench/mlos_bench/tests/optimizers/toy_optimization_loop_test.py
index f33a4f0983..c549133aa0 100644
--- a/mlos_bench/mlos_bench/tests/optimizers/toy_optimization_loop_test.py
+++ b/mlos_bench/mlos_bench/tests/optimizers/toy_optimization_loop_test.py
@@ -98,7 +98,7 @@ def test_mock_optimization_loop_no_defaults(
 def test_flaml_optimization_loop(mock_env_no_noise: MockEnv, flaml_opt: MlosCoreOptimizer) -> None:
     """Toy optimization loop with mock environment and FLAML optimizer."""
     (score, tunables) = _optimize(mock_env_no_noise, flaml_opt)
-    assert score == pytest.approx(60.15, 0.01)
+    assert score == pytest.approx(60.00, 0.01)
     assert tunables.get_param_values() == {
         "vmSize": "Standard_B2s",
         "idle": "halt",
diff --git a/mlos_core/setup.py b/mlos_core/setup.py
index b9ba6ac3c4..d86ced25fc 100644
--- a/mlos_core/setup.py
+++ b/mlos_core/setup.py
@@ -69,7 +69,7 @@ def _get_long_desc_from_readme(base_url: str) -> dict:
 
 extra_requires: dict[str, list[str]] = {  # pylint: disable=consider-using-namedtuple-or-dataclass
     "flaml": [
-        "flaml<2.2.0",  # FIXME: temporarily avoid changes in new FLAML package (#839).
+        "flaml>=2.3.3",
         "flaml[blendsearch]",
     ],
     # NOTE: Major refactoring on SMAC and ConfigSpace v1.0 starting from v2.2
diff --git a/setup.cfg b/setup.cfg
index 1c1cfe1416..2adc9d7660 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -52,6 +52,8 @@ filterwarnings =
     ignore:.*(get_hyperparam|get_dictionary|parents_of|to_vector|(list\(.*values\(\)\))).*:DeprecationWarning:smac:0
     ignore:.*(Please leave at default or explicitly set .size=None).*:DeprecationWarning:smac:0
     ignore:.*(declarative_base.*function is now available as sqlalchemy.orm.declarative_base):DeprecationWarning:optuna:0
+    ignore:.*(IntUniformDistribution.* is deprecated and internally converted to IntDistribution):FutureWarning:optuna:0
+    ignore:.*(flaml.automl is not available. Please install flaml\[automl\] to enable AutoML functionalities.):UserWarning:flaml:0
     ignore:.*(Trying to register a configuration that was not previously suggested).*:UserWarning:.*llamatune.*:0
     ignore:.*(DISPLAY environment variable is set).*:UserWarning:.*conftest.*:0
     ignore:.*(coroutine 'sleep' was never awaited).*:RuntimeWarning:.*event_loop_context_test.*:0