diff --git a/pyproject.toml b/pyproject.toml
index 20d97b43c..83fd465dd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -48,6 +48,7 @@ importlib-resources = { version = "*", python = "<3.9" }
 networkx = "^2.8.8"
 pywhy-graphs = { git = "https://github.com/py-why/pywhy-graphs.git", branch = 'main', optional = true }
 pygraphviz = { version = "*", optional = true }
+pyviz = { version = "*", optional = true}
 bnlearn = { git = "https://github.com/erdogant/bnlearn.git", branch = 'master', optional = true }
 
 [tool.poetry.group.style]