diff --git a/pyrovision/models/utils.py b/pyrovision/models/utils.py
index f451898..371e322 100644
--- a/pyrovision/models/utils.py
+++ b/pyrovision/models/utils.py
@@ -10,7 +10,7 @@
 
 import torch
 import torch.nn as nn
-from huggingface_hub import hf_hub_download
+from huggingface_hub.file_download import hf_hub_download
 
 from pyrovision import models