diff --git a/notebooks/validate_and_tokenize_data.ipynb b/notebooks/validate_and_tokenize_data.ipynb
index e4ecba5257..2ff3b56c55 100644
--- a/notebooks/validate_and_tokenize_data.ipynb
+++ b/notebooks/validate_and_tokenize_data.ipynb
@@ -52,7 +52,7 @@ dbutils.library.restartPython()
 
 # COMMAND ----------
 
-# MAGIC %pip install --upgrade --no-deps git+https://github.com/mosaicml/llm-foundry.git@byod/data_validation_fix
+# MAGIC %pip install --upgrade --no-deps git+https://github.com/mosaicml/llm-foundry.git@byod/data_validation
 # MAGIC %pip install 'mosaicml[libcloud,wandb,oci,gcs]>=0.23.4,<0.24'
 # MAGIC %pip install 'mlflow>=2.14.1,<2.16'
 # MAGIC %pip install 'transformers>=4.43.2,<4.44'