diff --git a/freqtrade/freqai/torch/PyTorchMLPModel.py b/freqtrade/freqai/torch/PyTorchMLPModel.py
index 192d2ad89..94cfa8e64 100644
--- a/freqtrade/freqai/torch/PyTorchMLPModel.py
+++ b/freqtrade/freqai/torch/PyTorchMLPModel.py
@@ -48,7 +48,7 @@ class PyTorchMLPModel(nn.Module):
         self.dropout = nn.Dropout(p=dropout_percent)
 
     def forward(self, x: List[torch.Tensor]) -> torch.Tensor:
-        x = x[0]
+        x: torch.Tensor = x[0]
         x = self.relu(self.input_layer(x))
         x = self.dropout(x)
         x = self.blocks(x)