diff --git a/validator/__init__.py b/validator/__init__.py
index 0ef309e..f2266a8 100644
--- a/validator/__init__.py
+++ b/validator/__init__.py
@@ -1,3 +1,3 @@
-from .main import DatasetEmbeddings
+from .main import ArizeDatasetEmbeddings
 
-__all__ = ["DatasetEmbeddings"]
+__all__ = ["ArizeDatasetEmbeddings"]
diff --git a/validator/arize_demo_dataset_embeddings_guard.ipynb b/validator/arize_demo_dataset_embeddings_guard.ipynb
index 9055e1b..31d1539 100644
--- a/validator/arize_demo_dataset_embeddings_guard.ipynb
+++ b/validator/arize_demo_dataset_embeddings_guard.ipynb
@@ -182,7 +182,7 @@
         "\n",
         "\n",
         "# can uncomment this once its public\n",
-        "!pip install  --no-cache-dir  -qq git+https://github.com/Arize-ai/jailbreak-embeddings\n",
+        "!pip install  --no-cache-dir  -qq git+https://github.com/Arize-ai/dataset-embeddings-guardrails\n",
         "\n",
         "from openinference.instrumentation.guardrails import GuardrailsInstrumentor\n",
         "from guardrails import Guard\n",