diff --git a/models/attention.py b/models/attention.py
index b535470..6f7f718 100755
--- a/models/attention.py
+++ b/models/attention.py
@@ -27,7 +27,7 @@
 import torch
 from torch import Tensor
 if float(torch.__version__.split('.')[0]) == 0 or (float(torch.__version__.split('.')[0]) == 1 and float(torch.__version__.split('.')[1])) < 9:
-    from torch.nn.modules.linear import _LinearWithBiass
+    from torch.nn.modules.linear import _LinearWithBias
 else:
     from torch.nn.modules.linear import NonDynamicallyQuantizableLinear as _LinearWithBias
 from torch.nn.init import xavier_uniform_