diff --git a/freqtrade/freqai/torch/PyTorchMLPModel.py b/freqtrade/freqai/torch/PyTorchMLPModel.py
index e50ccd5ef..62d3216df 100644
--- a/freqtrade/freqai/torch/PyTorchMLPModel.py
+++ b/freqtrade/freqai/torch/PyTorchMLPModel.py
@@ -2,7 +2,7 @@ import logging
 from typing import List
 
 import torch
-import torch.nn as nn
+from torch import nn
 
 
 logger = logging.getLogger(__name__)