diff --git a/models/Informer.py b/models/Informer.py
index 1d26d57..8abd5da 100644
--- a/models/Informer.py
+++ b/models/Informer.py
@@ -1,7 +1,7 @@
 import torch
 import torch.nn as nn
 from layers.Transformer_EncDec import Decoder, DecoderLayer, Encoder, EncoderLayer, ConvLayer
-from layers.SelfAttention_Family import ProbAttention, AttentionLayer
+from layers.SelfAttention_Family import ProbAttention, AttentionLayer, FullAttention
 from layers.Embed import DataEmbedding