diff --git a/src/alignment/model_utils.py b/src/alignment/model_utils.py
index 6795b6a..650517e 100644
--- a/src/alignment/model_utils.py
+++ b/src/alignment/model_utils.py
@@ -22,7 +22,7 @@ from transformers.trainer_utils import get_last_checkpoint
 
 from accelerate import Accelerator
 from huggingface_hub import list_repo_files
-from huggingface_hub.utils._errors import RepositoryNotFoundError
+from huggingface_hub.errors import RepositoryNotFoundError
 from huggingface_hub.utils._validators import HFValidationError
 from peft import LoraConfig, PeftConfig