diff --git a/pypots/imputation/autoformer/modules/core.py b/pypots/imputation/autoformer/modules/core.py
index ebe66f1f..d2d19f23 100644
--- a/pypots/imputation/autoformer/modules/core.py
+++ b/pypots/imputation/autoformer/modules/core.py
@@ -8,7 +8,6 @@
 import torch.nn as nn
 
 from .submodules import (
-    SeriesDecompositionBlock,
     SeasonalLayerNorm,
     AutoformerEncoderLayer,
     AutoformerEncoder,
@@ -38,7 +37,6 @@ def __init__(
 
         self.seq_len = n_steps
         self.n_layers = n_layers
-        self.series_decomp = SeriesDecompositionBlock(moving_avg_window_size)
         self.enc_embedding = DataEmbedding(
             n_features,
             d_model,
@@ -59,7 +57,7 @@ def __init__(
                     dropout,
                     activation,
                 )
-                for i in range(n_layers)
+                for _ in range(n_layers)
             ],
             norm_layer=SeasonalLayerNorm(d_model),
         )