diff --git a/pyproject.toml b/pyproject.toml
index 94e27681..1fcd31f0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,7 +31,7 @@ packages = ["tsfm_public", "tsfmhfdemos"]
 [project.optional-dependencies]
 
 
-all = ["tsfm_public[notebooks,testing,dev]"]
+all = ["granite-tsfm[notebooks,testing,dev]"]
 
 notebooks = [
     "jupyter",
@@ -42,14 +42,14 @@ notebooks = [
     "kaleido",
     "tensorboard",
 ]
-testing = ["pytest", "tsfm_public[notebooks]", "parameterized"]
-dev = ["pre-commit", "tsfm_public[testing]", "ruff==0.4.4"]
+testing = ["pytest", "granite-tsfm[notebooks]", "parameterized"]
+dev = ["pre-commit", "granite-tsfm[testing]", "ruff==0.4.4"]
 
 # ogv deployments will already have jupyter
 # and we don't want to mess with RH's tested version
 ogv = ["datasets", "plotly", "kaleido"]
 demos = [
-    "tsfm_public[notebooks]",
+    "granite-tsfm[notebooks]",
     "streamlit",
     "altair<5",
     "plotly",