diff --git a/setup.py b/setup.py
index 9b93e0c903..e9150264b6 100644
--- a/setup.py
+++ b/setup.py
@@ -108,9 +108,7 @@ def dmf_data_files(root: str = DMF_DATA_ROOT) -> List[Tuple[str, List[str]]]:
         "prerelease": DEPENDENCIES_FOR_PRERELEASE_VERSION,
         "optional": [
             "tensorflow",  # idaes.surrogate.keras_surrogate
-            # FIXME update requirement once PyPI distribution is updated
-            # (8fef0fa2ca is the commit on main where #128 was merged in)
-            "gridx-prescient @ https://github.com/grid-parity-exchange/Prescient/archive/8fef0fa2ca.zip",  # idaes.tests.prescient
+            "gridx-prescient==2.1",  # idaes.tests.prescient
             # A Lee 11-Jan-22: no precompiled version of CoolProp available for Pyhton 3.9
             "coolprop; python_version < '3.9'",  # idaes.generic_models.properties.general.coolprop
         ],