diff --git a/test/test_hub.py b/test/test_hub.py
index 2f0ddfb2537..4ae9e51021b 100644
--- a/test/test_hub.py
+++ b/test/test_hub.py
@@ -44,7 +44,7 @@ def test_set_dir(self):
             progress=False)
         self.assertEqual(sum_of_model_parameters(hub_model).item(),
                          SUM_OF_PRETRAINED_RESNET18_PARAMS)
-        assert os.path.exists(temp_dir + '/pytorch_vision_master')
+        self.assertTrue(os.path.exists(temp_dir + '/pytorch_vision_master'))
         shutil.rmtree(temp_dir + '/pytorch_vision_master')
 
     def test_list_entrypoints(self):