diff --git a/tests/test_common/test_utils/test_checkpoints.py b/tests/test_common/test_utils/test_checkpoints.py
index 66dabba..bd58fe5 100644
--- a/tests/test_common/test_utils/test_checkpoints.py
+++ b/tests/test_common/test_utils/test_checkpoints.py
@@ -5,7 +5,6 @@
 import torch
 import torch.optim as optim
 import torchvision.models as models
-from torch.nn.modules.module import T
 
 from code_soup.common.utils.checkpoints import Checkpoints