diff --git a/chatdocs/document_loaders/nougat_loader.py b/chatdocs/document_loaders/nougat_loader.py
index 201512e..46a5096 100644
--- a/chatdocs/document_loaders/nougat_loader.py
+++ b/chatdocs/document_loaders/nougat_loader.py
@@ -34,7 +34,7 @@ def __init__(
                 "`nougat` package not found, please install it with "
                 "`pip install nougat-ocr`"
             )
-        checkpoint = get_checkpoint("nougat", download=True)
+        checkpoint = get_checkpoint(download=True)
         self.model = NougatModel.from_pretrained(checkpoint)
 
         if torch.cuda.is_available():