diff --git a/mlos_core/setup.py b/mlos_core/setup.py
index 2dd57d064e..d856490b67 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.4.0",  # FIXME: temporarily avoid changes in new FLAML package (#839).
         "flaml[blendsearch]",
     ],
     # NOTE: Major refactoring on SMAC and ConfigSpace v1.0 starting from v2.2