diff --git a/models/Informer.py b/models/Informer.py
index 916c0ab..1d26d57 100644
--- a/models/Informer.py
+++ b/models/Informer.py
@@ -49,7 +49,7 @@ def __init__(self, configs):
                         ProbAttention(True, configs.factor, attention_dropout=configs.dropout, output_attention=False),
                         configs.d_model, configs.n_heads),
                     AttentionLayer(
-                        ProbAttention(False, configs.factor, attention_dropout=configs.dropout, output_attention=False),
+                        FullAttention(False, configs.factor, attention_dropout=configs.dropout, output_attention=False),
                         configs.d_model, configs.n_heads),
                     configs.d_model,
                     configs.d_ff,