diff --git a/timesformer/models/resnet_helper.py b/timesformer/models/resnet_helper.py
index 082d318..773602b 100755
--- a/timesformer/models/resnet_helper.py
+++ b/timesformer/models/resnet_helper.py
@@ -12,7 +12,6 @@
 from einops import rearrange, reduce, repeat
 import torch.nn.functional as F
 from torch.nn.modules.module import Module
-from torch.nn.modules.linear import _LinearWithBias
 from torch.nn.modules.activation import MultiheadAttention
 
 import numpy as np