diff --git a/setup.cfg b/setup.cfg
index f672af25..2df26759 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -112,6 +112,7 @@ deps =
 	palmerpenguins
 	Pillow
 	polars
+ 	prefect<=2.20.3,>=2.16.5
 	pytest
 	pytest-cov
 	xgboost