diff --git a/pyproject.toml b/pyproject.toml
index 4f90038..118c5f4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -49,7 +49,7 @@ doc = [
 ]
 
 dev = [
-  "taolib",
+  "taolib[flows]",
   "onnx",
   "onnxscript",
   "onnxruntime",