diff --git a/tests/test_efficientnet.py b/tests/test_efficientnet.py
index d36157e6fa..f8b14beada 100644
--- a/tests/test_efficientnet.py
+++ b/tests/test_efficientnet.py
@@ -13,6 +13,7 @@
 import unittest
 from typing import TYPE_CHECKING
 from unittest import skipUnless
+from urllib.error import ContentTooShortError, HTTPError
 
 import torch
 from parameterized import parameterized