diff --git a/hi-ml-multimodal/test_multimodal/image/model/test_model.py b/hi-ml-multimodal/test_multimodal/image/model/test_model.py
index 36f17eb33..c8267c567 100644
--- a/hi-ml-multimodal/test_multimodal/image/model/test_model.py
+++ b/hi-ml-multimodal/test_multimodal/image/model/test_model.py
@@ -143,7 +143,7 @@ def test_hubconf() -> None:
     """Test that instantiating the image model using the PyTorch Hub is consistent with older methods."""
     image = torch.rand(1, 3, 480, 480)
 
-    github = 'microsoft/hi-ml'
+    github = 'microsoft/hi-ml:main'
     model_hub = torch.hub.load(github, 'biovil_resnet', pretrained=True)
     model_himl = get_biovil_resnet()