diff --git a/kws/model/loss.py b/kws/model/loss.py
index 450cd95..eac9597 100644
--- a/kws/model/loss.py
+++ b/kws/model/loss.py
@@ -13,7 +13,6 @@
 # limitations under the License.
 
 import torch
-import torch.nn as nn
 import torch.nn.functional as F
 
 from kws.utils.mask import padding_mask