diff --git a/src/resources/scripts/ExtractFeatures.py b/src/resources/scripts/ExtractFeatures.py
index 1074741..6df644f 100644
--- a/src/resources/scripts/ExtractFeatures.py
+++ b/src/resources/scripts/ExtractFeatures.py
@@ -19,7 +19,7 @@
 with open(sys.argv[1], 'r') as f:
     input_json = json.load(f)
 
-dinov2_vits14 = hub_load('facebookresearch/dinov2', 'dinov2_vits14')
+dinov2_vits14 = hub_load('facebookresearch/dinov2:main', 'dinov2_vits14')
 
 if cuda_is_available():
     device = device('cuda')